From e24cf595619a752d8b2363260a3d5b1545488d61 Mon Sep 17 00:00:00 2001
From: TiantianLi28 <102604240+TiantianLi28@users.noreply.github.com>
Date: Mon, 4 Nov 2024 11:57:33 +0800
Subject: [PATCH] Update scatter_points.py

added input variables
---
 mmcv/ops/scatter_points.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/mmcv/ops/scatter_points.py b/mmcv/ops/scatter_points.py
index e0aec67ce6..39def43e08 100644
--- a/mmcv/ops/scatter_points.py
+++ b/mmcv/ops/scatter_points.py
@@ -39,11 +39,12 @@ def forward(ctx: Any,
         ctx.device = feats.device.type
         if ctx.device == 'npu':
             import ads_c
-            voxel_idx = ads_c.point_to_voxel(coors, [], [])
+            voxel_idx = ads_c.point_to_voxel(coors, [], [], 'XYZ')
             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, [], [])
+            voxel_coors = ads_c.voxel_to_point(uniqued_voxel_idx,
+                                               [], [], 'XYZ')
             voxel_feats, \
                 compare_mask = ads_c.npu_dynamic_scatter(feats, coors,
                                                          prefix_sum,