NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +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
{
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