Skip to content

Commit f1504cc

Browse files
authored
fix import - again
1 parent 6bb58f8 commit f1504cc

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

numba_cuda/numba/cuda/cudadrv/driver.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,11 @@
6161
Stream as ExperimentalStream,
6262
Device as ExperimentalDevice,
6363
)
64-
from cuda.core.graph import GraphBuilder
6564
from numba.cuda._compat import CUDA_CORE_GT_0_6, CUDA_CORE_GE_1_0
65+
if CUDA_CORE_GE_1_0:
66+
from cuda.core.graph import GraphBuilder
67+
else:
68+
from cuda.core import GraphBuilder
6669
from cuda.bindings.utils import get_cuda_native_handle
6770

6871

0 commit comments

Comments
 (0)