Merge pull request PolusAI#209 from JesseMckinzie/gpu_fixes

Bug fixes
sameeul · Mar 5, 2024 · 765db72 · 765db72
2 parents 154c164 + 933f15e
commit 765db72
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/nyx/environment_basic.cpp b/src/nyx/environment_basic.cpp
@@ -15,6 +15,11 @@
 BasicEnvironment::BasicEnvironment()
 {
 	temp_dir_path = fs::temp_directory_path().string();
+
+	// Add slash to path if needed
+	if (!temp_dir_path.empty() && temp_dir_path.back() != fs::path::preferred_separator) {
+        temp_dir_path += fs::path::preferred_separator;
+    }
 }
 
 bool BasicEnvironment::check_2d_file_pattern(const std::string& pat)

diff --git a/src/nyx/python/new_bindings_py.cpp b/src/nyx/python/new_bindings_py.cpp
@@ -104,6 +104,7 @@ void initialize_environment(
             theEnvironment.set_use_gpu(false);
         } else {
             theEnvironment.set_gpu_device_id(using_gpu);
+            theEnvironment.set_use_gpu(true);
         }
     #else 
         if (using_gpu != -1) {