Skip to content

GitLab

Unverified Commit 74b5267d authored by Jane (Yuan) Xu's avatar Jane (Yuan) Xu Committed by GitHub
Browse files

Use narrow over indexing in `hadacore_transform` to prep for ABI stable (#28756) 


Signed-off-by: default avatarJane Xu <janeyx@meta.com>
parent dd6ac1c2
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu
View file @ 74b5267d
...@@ -802,7 +802,7 @@ torch::Tensor hadacore_transform(torch::Tensor& x, bool inplace) { ...@@ -802,7 +802,7 @@ torch::Tensor hadacore_transform(torch::Tensor& x, bool inplace) {
}); });
if (numel % 256 != 0) { 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()) { if (inplace && out.data_ptr() != x.data_ptr()) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment