Skip to content

Commit 5d4b515

Browse files
committed
*correctly* fix the default cuda context init in pycuda accuracy tests...
1 parent ac4b0b4 commit 5d4b515

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyvkfft/accuracy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def init_ctx(backend, gpu_name=None, opencl_platform=None, verbose=False):
8989
else:
9090
raise RuntimeError("Selected backend is pycuda, but no device found")
9191
gpu_ctx_dic["pycuda"] = (d, d.retain_primary_context())
92-
gpu_ctx_dic["pycuda"].push()
92+
gpu_ctx_dic["pycuda"][1].push()
9393
if verbose:
9494
print("Selected device for pycuda: %s" % d.name())
9595
elif backend == "pyopencl":

0 commit comments

Comments
 (0)