Use narrow over indexing in hadacore_transform to prep for ABI stable (vllm-project#28756)

janeyx99 · geodavic · commit a74e52f8afbf · 2025-11-15T20:55:24.000-06:00
Signed-off-by: Jane Xu &lt;janeyx@meta.com&gt;
Signed-off-by: George D. Torres &lt;gdavtor@gmail.com&gt;
diff --git a/csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu b/csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu
@@ -802,7 +802,7 @@ torch::Tensor hadacore_transform(torch::Tensor& x, bool inplace) {
     });
 
     if (numel % 256 != 0) {
-        out = out.index({torch::indexing::Slice(0, numel / had_size)});
+        out = out.narrow(0, 0, numel / had_size);
     }
 
     if (inplace && out.data_ptr() != x.data_ptr()) {

Original file line number	Diff line number	Diff line change
`@@ -802,7 +802,7 @@ torch::Tensor hadacore_transform(torch::Tensor& x, bool inplace) {`
`802`	`802`	`});`
`803`	`803`
`804`	`804`	`if (numel % 256 != 0) {`
`805`		`- out = out.index({torch::indexing::Slice(0, numel / had_size)});`
	`805`	`+ out = out.narrow(0, 0, numel / had_size);`
`806`	`806`	`}`
`807`	`807`
`808`	`808`	`if (inplace && out.data_ptr() != x.data_ptr()) {`