Update cuda_core/tests/conftest.py

Co-authored-by: Leo Fang <leof@nvidia.com>

Author: ksimpson-work

cuda_core/tests/conftest.py

@@ -20,13 +20,14 @@ def init_cuda():
     device = Device()
     device.set_current()
     yield
+    _device_unset_current()
+
+def _device_unset_current():
     handle_return(driver.cuCtxPopCurrent())
     with _device._tls_lock:
         del _device._tls.devices
 
 @pytest.fixture(scope="function")
 def deinit_cuda():
     yield
-    handle_return(driver.cuCtxPopCurrent())
-    with _device._tls_lock:
-        del _device._tls.devices
+    _device_unset_current()