Merge pull request #51 from DaGaiBa/rc4main

Interfaces change.
momo609 · Jul 19, 2024 · 44afa6a · 44afa6a
2 parents e112ca4 + f572475
commit 44afa6a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmcv/ops/scatter_points.py b/mmcv/ops/scatter_points.py
@@ -39,11 +39,11 @@ def forward(ctx: Any,
         ctx.device = feats.device.type
         if ctx.device == 'npu':
             import ads_c
-            voxel_idx = ads_c.point_to_voxel(coors, None, None)
+            voxel_idx = ads_c.point_to_voxel(coors, [], [])
             unique_res = ads_c.unique_voxel(voxel_idx)
             num_voxels, uniqued_voxel_idx, prefix_sum, \
-                argsort_coor = unique_res
-            voxel_coors = ads_c.voxel_to_point(uniqued_voxel_idx, None, None)
+                argsort_coor, _ = unique_res
+            voxel_coors = ads_c.voxel_to_point(uniqued_voxel_idx, [], [])
             voxel_feats, \
                 compare_mask = ads_c.npu_dynamic_scatter(feats, coors,
                                                          prefix_sum,