Include CUDA_VISIBLE_DEVICES into shm hash, so same db can be loaded …

…into multiple GPUs
soedinglab · Jan 18, 2025 · b804fbe · b804fbe
1 parent e095774
commit b804fbe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/commons/GpuUtil.h b/src/commons/GpuUtil.h
@@ -41,6 +41,10 @@ struct GPUSharedMemory {
 
     static std::string getShmHash(const std::string& db) {
         std::string dbpath = FileUtil::getRealPathFromSymLink(PrefilteringIndexReader::dbPathWithoutIndex(db));
+        char* visibleDevices = getenv("CUDA_VISIBLE_DEVICES");
+        if (visibleDevices) {
+            dbpath.append(visibleDevices);
+        }
         size_t hash = Util::hash(dbpath.c_str(), dbpath.length());
         return SSTR(hash);
     }