From c747a19b98a3d24382b213177f07ddcc23bffe2c Mon Sep 17 00:00:00 2001 From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com> Date: Fri, 25 Jul 2025 12:42:58 -0500 Subject: [PATCH] Remove code setting up CUDA_MODULE_LOADING=EAGER in Python extension --- python/src/py_nvbench.cpp | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp index c93581f..34c812d 100644 --- a/python/src/py_nvbench.cpp +++ b/python/src/py_nvbench.cpp @@ -38,15 +38,6 @@ namespace py = pybind11; namespace { -inline void set_env(const char *name, const char *value) -{ -#ifdef _MSC_VER - _putenv_s(name, value); -#else - setenv(name, value, 1); -#endif -} - struct PyObjectDeleter { void operator()(py::object *p) @@ -220,9 +211,6 @@ PYBIND11_MODULE(_nvbench, m) // == STEP 1 // Set environment variable CUDA_MODULE_LOADING=EAGER - // See NVIDIA/NVBench#136 for CUDA_MODULE_LOADING - set_env("CUDA_MODULE_LOADING", "EAGER"); - NVBENCH_DRIVER_API_CALL(cuInit(0)); // This line ensures that benchmark_manager has been created during module init