Skip to content

Commit cb1a1b2

Browse files
committed
Found a missed instance of cuda.
1 parent 31f2465 commit cb1a1b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernel_tuner/backends/nvcuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def __init__(self, device=0, iterations=7, compiler_options=None, observers=None
4747
# initialize and select device
4848
err = driver.cuInit(0)
4949
cuda_error_check(err)
50-
err, self.device = cuda.cuDeviceGet(device)
50+
err, self.device = driver.cuDeviceGet(device)
5151
cuda_error_check(err)
5252
err, self.context = driver.cuDevicePrimaryCtxRetain(device)
5353
cuda_error_check(err)

0 commit comments

Comments
 (0)