NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-30 03:31:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove code setting up CUDA_MODULE_LOADING=EAGER in Python extension

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-25 12:42:58 -05:00
parent 985db4f144
commit c747a19b98
1 changed files with 0 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/src/py_nvbench.cpp
View File

@@ -38,15 +38,6 @@ namespace py = pybind11;
namespace 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 struct PyObjectDeleter
{ {
void operator()(py::object *p) void operator()(py::object *p)
@@ -220,9 +211,6 @@ PYBIND11_MODULE(_nvbench, m)
// == STEP 1 // == STEP 1
// Set environment variable CUDA_MODULE_LOADING=EAGER // 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)); NVBENCH_DRIVER_API_CALL(cuInit(0));
// This line ensures that benchmark_manager has been created during module init // This line ensures that benchmark_manager has been created during module init