Skip to content

Commit 3b58580

Browse files
author
Shawn Lin
committed
small fixes
1 parent 019e780 commit 3b58580

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎sumpy/p2p.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -754,6 +754,7 @@ def get_optimized_kernel(self, *,
754754
if strength_dtype == source_dtype:
755755
knl = lp.concatenate_arrays(knl, local_arrays, "local_isrc")
756756
local_arrays = ["local_isrc"]
757+
local_array_isrc_axis = [1]
757758
local_array_sizes = [self.dim + self.strength_count]
758759
local_array_dtypes = [source_dtype]
759760
# We try to mark the local arrays (sources, strengths)
@@ -811,8 +812,8 @@ def __call__(self,
811812

812813
is_gpu = not is_cl_cpu(actx)
813814
if is_gpu:
814-
source_dtype = kwargs["sources"][0].dtype
815-
strength_dtype = kwargs["strength"].dtype
815+
source_dtype = sources[0].dtype
816+
strength_dtype = kwargs["strength"][0].dtype
816817
else:
817818
# these are unused for not GPU and defeats the caching
818819
# set them to None to keep the caching across dtypes

0 commit comments

Comments
 (0)