From f5724756a7aaed9123ca9d660bcb2de296ebae8e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=8C=AF=E8=B1=AA?= Date: Fri, 19 Jul 2024 16:09:37 +0800 Subject: [PATCH] Interfaces change. --- mmcv/ops/scatter_points.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mmcv/ops/scatter_points.py b/mmcv/ops/scatter_points.py index 6902343a03..e0aec67ce6 100644 --- 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,