From 4e8972fbf9be5790523950af6edebb4a2ab40ee0 Mon Sep 17 00:00:00 2001 From: liuhw <71241939+dflhw@users.noreply.github.com> Date: Mon, 20 Mar 2023 22:08:30 +0800 Subject: [PATCH] Move voxel_size to the device where points are (#2681) * ... * ... * ... --- mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp b/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp index 13e50401f..2b22646b9 100644 --- a/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp @@ -24,13 +24,11 @@ int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels, at::Tensor voxel_size_cpu = at::from_blob( const_cast(voxel_size.data()), {3}, dtype(at::kFloat)); - at::Tensor voxel_size_npu = - CalcuOpUtil::CopyTensorHostToDevice(voxel_size_cpu); + at::Tensor voxel_size_npu = voxel_size_cpu.to(points.device()); at::Tensor coors_range_cpu = at::from_blob( const_cast(coors_range.data()), {6}, dtype(at::kFloat)); - at::Tensor coors_range_npu = - CalcuOpUtil::CopyTensorHostToDevice(coors_range_cpu); + at::Tensor coors_range_npu = coors_range_cpu.to(points.device()); int64_t max_points_ = (int64_t)max_points; int64_t max_voxels_ = (int64_t)max_voxels;