Portability for setenv.

NVIDIA · Apr 4, 2024 · d77f335 · d77f335
1 parent 868d951
commit d77f335
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nvbench/main.cuh b/nvbench/main.cuh
@@ -31,7 +31,11 @@
 static int _nvbench_env [[maybe_unused]] = []() -> int
 {
   // See NVIDIA/NVBench#136
+#ifdef _MSC_VER
+  int retval = _putenv_s("CUDA_MODULE_LOADING", "EAGER");
+#else
   int retval = setenv("CUDA_MODULE_LOADING", "EAGER", 1);
+#endif
   return retval;
 }();