[Enhance] Do not use deprecated .type().scalarType() (#3195)

`at::Tensor::type()` method is deprecated and `tensor.scalar_type()` is a recommended way to replace `tensor.type().scalarType()` invocation
open-mmlab · Nov 4, 2024 · 71437a3 · 71437a3
1 parent 1393257
commit 71437a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/spconv_utils.h b/mmcv/ops/csrc/pytorch/spconv_utils.h
@@ -35,7 +35,7 @@ struct TorchGPU : public tv::GPU {
 
 template <typename scalar_t>
 void check_torch_dtype(const torch::Tensor &tensor) {
-  switch (tensor.type().scalarType()) {
+  switch (tensor.scalar_type()) {
     case at::ScalarType::Double: {
       auto val = std::is_same<std::remove_const_t<scalar_t>, double>::value;
       TV_ASSERT_RT_ERR(val, "error");