Skip to content

Commit f28dfe2

Browse files
committed
Generate Numba signature objects when Numba is in use
1 parent a6899c5 commit f28dfe2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

numba_cuda/numba/cuda/typing/templates.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,11 @@ def generic(self):
200200
def signature(return_type, *args, **kws):
201201
recvr = kws.pop("recvr", None)
202202
assert not kws
203-
return Signature(return_type, args, recvr=recvr)
203+
if HAS_NUMBA:
204+
signature_class = CoreSignature
205+
else:
206+
signature_class = Signature
207+
return signature_class(return_type, args, recvr=recvr)
204208

205209

206210
def fold_arguments(

0 commit comments

Comments
 (0)