From 294c0907aa5c8f0b59abb845c4941703d34cfcc9 Mon Sep 17 00:00:00 2001 From: Hesam Salehipour Date: Wed, 28 May 2025 11:01:29 -0400 Subject: [PATCH] Fixed a couple bugs in bindings --- py/neon/dense/dPartition.h | 2 +- py/neon/dense/dPartition.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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,