diff --git a/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp b/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp index 32b1a50cc1..ffd9b4c43b 100644 --- a/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp @@ -18,7 +18,7 @@ int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels, const std::vector coors_range, const int max_points, const int max_voxels, const int NDim = 3) { - at::Tensor voxel_num_tmp = OpPreparation::ApplyTensor(points, {1}); + at::Tensor voxel_num_tmp = at::empty({1}, points.options()); at::Tensor voxel_num = voxel_num_tmp.to(at::kInt); at::Tensor voxel_size_cpu = at::from_blob(