Skip to content
This repository was archived by the owner on Jan 25, 2023. It is now read-only.

Commit 3d663cf

Browse files
authored
Extending typing context (#101)
Changed in 7ccc8e3 by @reazulhoque.
1 parent 22a29d9 commit 3d663cf

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

numba/core/typing/context.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -352,17 +352,8 @@ def resolve_argument_type(self, val):
352352
try:
353353
return typeof(val, Purpose.argument)
354354
except ValueError:
355-
from numba.dppl_config import dppl_present, DeviceArray
356-
if dppl_present:
357-
if(type(val) == DeviceArray):
358-
return typeof(val._ndarray, Purpose.argument)
359-
# DRD : Hmmm... is the assumption that this error is encountered
360-
# when someone is using cuda, and already has done an import
361-
# cuda?
362-
#elif numba.cuda.is_cuda_array(val):
363-
# return typeof(numba.cuda.as_cuda_array(val), Purpose.argument)
364-
else:
365-
raise
355+
if numba.cuda.is_cuda_array(val):
356+
return typeof(numba.cuda.as_cuda_array(val), Purpose.argument)
366357
else:
367358
raise
368359

0 commit comments

Comments
 (0)