frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Abate -Wsign-compare warning spam in Indexing.cu (#134805)

Browse Source 
Fix for warning spam like
```
 warning: comparison of integer expressions of different signedness: ‘long int’ and ‘uint64_t’ {aka ‘long unsigned int’} [-Wsign-compare]
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134805
Approved by: https://github.com/janeyx99
This commit is contained in:
eqy
2024-09-01 10:48:04 +00:00
committed by PyTorch MergeBot
parent d03f767cae
commit b1a00b7b6d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
aten/src/ATen/native/cuda/Indexing.cu
View File

@ -1350,7 +1350,7 @@ template <typename scalar_t>
void index_select_out_cuda_impl(
Tensor& out,
const Tensor& self,
long dim,
uint64_t dim,
const Tensor& index) {
uint64_t numIndices = index.numel();
uint64_t selfDims = self.dim() == 0 ? 1 : self.dim();
@ -1511,13 +1511,13 @@ Tensor& index_select_out_cuda(
self.qscheme() == kPerTensorAffine,
"Only per_tensor quantized quantized tensors are supported by index_select.")
AT_DISPATCH_QINT_TYPES(out.scalar_type(), "index_select_quant_cuda", [&] {
index_select_out_cuda_impl<scalar_t>(out, self, dim, index);
index_select_out_cuda_impl<scalar_t>(out, self, (uint64_t) dim, index);
});
} else {
AT_DISPATCH_V2(
out.scalar_type(),
"index_select_cuda",
AT_WRAP([&] { index_select_out_cuda_impl<scalar_t>(out, self, dim, index); }),
AT_WRAP([&] { index_select_out_cuda_impl<scalar_t>(out, self, (uint64_t) dim, index); }),
AT_EXPAND(AT_ALL_TYPES_AND_COMPLEX), AT_EXPAND(AT_BAREBONES_UNSIGNED_TYPES),
kComplexHalf,
kHalf,