Skip to content

Commit e709063

Browse files
Remove code setting up CUDA_MODULE_LOADING=EAGER in Python extension
1 parent a572ed1 commit e709063

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

python/src/py_nvbench.cpp

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -38,15 +38,6 @@ namespace py = pybind11;
3838
namespace
3939
{
4040

41-
inline void set_env(const char *name, const char *value)
42-
{
43-
#ifdef _MSC_VER
44-
_putenv_s(name, value);
45-
#else
46-
setenv(name, value, 1);
47-
#endif
48-
}
49-
5041
struct PyObjectDeleter
5142
{
5243
void operator()(py::object *p)
@@ -220,9 +211,6 @@ PYBIND11_MODULE(_nvbench, m)
220211
// == STEP 1
221212
// Set environment variable CUDA_MODULE_LOADING=EAGER
222213

223-
// See NVIDIA/NVBench#136 for CUDA_MODULE_LOADING
224-
set_env("CUDA_MODULE_LOADING", "EAGER");
225-
226214
NVBENCH_DRIVER_API_CALL(cuInit(0));
227215

228216
// This line ensures that benchmark_manager has been created during module init

0 commit comments

Comments
 (0)