diff --git a/py/neon/dense/dPartition.h b/py/neon/dense/dPartition.h index c9a28623..966b97ad 100644 --- a/py/neon/dense/dPartition.h +++ b/py/neon/dense/dPartition.h @@ -121,7 +121,7 @@ CUDA_CALLABLE inline auto neon_cardinality( } template -CUDA_CALLABLE inline auto neon_ngh_data( +CUDA_CALLABLE inline auto neon_read_ngh( const NeonDensePartition& p, NeonDenseIdx const & idx, NeonNghIdx const & ngh, diff --git a/py/neon/dense/dPartition.py b/py/neon/dense/dPartition.py index a2dd2554..26b14332 100644 --- a/py/neon/dense/dPartition.py +++ b/py/neon/dense/dPartition.py @@ -181,7 +181,7 @@ def register_builtins(): ) wp.context.add_builtin( - "neon_ngh_data", + "neon_read_ngh", input_types={"partition": Partition, 'idx': dIndex, 'ngh_idx': Ngh_idx,