There was an error while loading. Please reload this page.
1 parent 6bb58f8 commit f1504ccCopy full SHA for f1504cc
1 file changed
numba_cuda/numba/cuda/cudadrv/driver.py
@@ -61,8 +61,11 @@
61
Stream as ExperimentalStream,
62
Device as ExperimentalDevice,
63
)
64
-from cuda.core.graph import GraphBuilder
65
from numba.cuda._compat import CUDA_CORE_GT_0_6, CUDA_CORE_GE_1_0
+if CUDA_CORE_GE_1_0:
66
+ from cuda.core.graph import GraphBuilder
67
+else:
68
+ from cuda.core import GraphBuilder
69
from cuda.bindings.utils import get_cuda_native_handle
70
71
0 commit comments