"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "5d1372c0251d9b961e1d550ced8a07260426ff30"
Unverified Commit 4e8972fb authored by liuhw's avatar liuhw Committed by GitHub
Browse files

Move voxel_size to the device where points are (#2681)

* ...

* ...

* ...
parent 06fa3285
...@@ -24,13 +24,11 @@ int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels, ...@@ -24,13 +24,11 @@ int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels,
at::Tensor voxel_size_cpu = at::from_blob( at::Tensor voxel_size_cpu = at::from_blob(
const_cast<float *>(voxel_size.data()), {3}, dtype(at::kFloat)); const_cast<float *>(voxel_size.data()), {3}, dtype(at::kFloat));
at::Tensor voxel_size_npu = at::Tensor voxel_size_npu = voxel_size_cpu.to(points.device());
CalcuOpUtil::CopyTensorHostToDevice(voxel_size_cpu);
at::Tensor coors_range_cpu = at::from_blob( at::Tensor coors_range_cpu = at::from_blob(
const_cast<float *>(coors_range.data()), {6}, dtype(at::kFloat)); const_cast<float *>(coors_range.data()), {6}, dtype(at::kFloat));
at::Tensor coors_range_npu = at::Tensor coors_range_npu = coors_range_cpu.to(points.device());
CalcuOpUtil::CopyTensorHostToDevice(coors_range_cpu);
int64_t max_points_ = (int64_t)max_points; int64_t max_points_ = (int64_t)max_points;
int64_t max_voxels_ = (int64_t)max_voxels; int64_t max_voxels_ = (int64_t)max_voxels;
......
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