*correctly* fix the default cuda context init in pycuda accuracy test…

…s...
vincefn · Feb 13, 2024 · 5d4b515 · 5d4b515
1 parent ac4b0b4
commit 5d4b515
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyvkfft/accuracy.py b/pyvkfft/accuracy.py
@@ -89,7 +89,7 @@ def init_ctx(backend, gpu_name=None, opencl_platform=None, verbose=False):
             else:
                 raise RuntimeError("Selected backend is pycuda, but no device found")
         gpu_ctx_dic["pycuda"] = (d, d.retain_primary_context())
-        gpu_ctx_dic["pycuda"].push()
+        gpu_ctx_dic["pycuda"][1].push()
         if verbose:
             print("Selected device for pycuda: %s" % d.name())
     elif backend == "pyopencl":