From 6858a3ad5591628361a72de5496a54ae742f0a97 Mon Sep 17 00:00:00 2001 From: Ian Thomas Date: Wed, 7 Dec 2022 13:54:55 +0000 Subject: [PATCH] Ensure cuda functions are correctly jitted --- datashader/reductions.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/datashader/reductions.py b/datashader/reductions.py index 4aec1c14c..91c18c23d 100644 --- a/datashader/reductions.py +++ b/datashader/reductions.py @@ -662,7 +662,7 @@ def _append(x, y, agg, field): pass @staticmethod - @ngjit + @nb_cuda.jit(device=True) def _append_cuda(x, y, agg, field): # not called, the upsample function must set agg directly pass @@ -725,7 +725,7 @@ def _append_antialias_not_self_intersect(x, y, agg, field, aa_factor): # GPU append functions @staticmethod - @ngjit + @nb_cuda.jit(device=True) def _append_cuda(x, y, agg, field): if not isnull(field): nb_cuda.atomic.add(agg, (y, x), field) @@ -906,7 +906,7 @@ def _append_antialias(x, y, agg, field, aa_factor): # GPU append functions @staticmethod - @ngjit + @nb_cuda.jit(device=True) def _append_cuda(x, y, agg, field): cuda_atomic_nanmin(agg, (y, x), field) @@ -945,12 +945,12 @@ def _append_antialias(x, y, agg, field, aa_factor): # GPU append functions @staticmethod - @ngjit + @nb_cuda.jit(device=True) def _append_antialias_cuda(x, y, agg, field, aa_factor): cuda_atomic_nanmax(agg, (y, x), field*aa_factor) @staticmethod - @ngjit + @nb_cuda.jit(device=True) def _append_cuda(x, y, agg, field): cuda_atomic_nanmax(agg, (y, x), field)