Skip to content

Commit 3cb8a8d

Browse files
committed
add roi_align_rotated npu adpater
1 parent f2439cd commit 3cb8a8d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmcv/ops/csrc/pytorch/npu/voxelization_npu.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ int hard_voxelize_forward_npu(const at::Tensor &points, at::Tensor &voxels,
1818
const std::vector<float> coors_range,
1919
const int max_points, const int max_voxels,
2020
const int NDim = 3) {
21-
at::Tensor voxel_num_tmp = OpPreparation::ApplyTensor(points, {1});
21+
at::Tensor voxel_num_tmp = at::empty({1}, points.options());
2222
at::Tensor voxel_num = voxel_num_tmp.to(at::kInt);
2323

2424
at::Tensor voxel_size_cpu = at::from_blob(

0 commit comments

Comments
 (0)