1 files changed, 14 insertions, 5 deletions
diff --git a/source/core/slang-test-tool-util.cpp b/source/core/slang-test-tool-util.cpp
index 68673a29f..487ea89ac 100644
--- a/source/core/slang-test-tool-util.cpp
+++ b/source/core/slang-test-tool-util.cpp
@@ -103,12 +103,21 @@ static SlangResult _addCUDAPrelude(const String& rootPath, slang::IGlobalSession
     String parentPath;
     SLANG_RETURN_ON_FAIL(getExeDirectoryPath(inExePath, parentPath));
 
-    // From directory to the root is ../../.. 
-    // Work out the relative path to the root
-    String rootRelPath = Path::combine(parentPath, "../../../");
+    // Work out the relative path to the root, we will search upwards until we
+    // find a directory containing 'prelude/slang-cpp-prelude.h'
+    String rootRelPath;
+    SLANG_RETURN_ON_FAIL(Path::getCanonical(parentPath, rootRelPath));
+    do
+    {
+        if(File::exists(Path::combine(rootRelPath, "prelude/slang-cpp-prelude.h")))
+            break;
+
+        rootRelPath = Path::getParentDirectory(rootRelPath);
+        if(rootRelPath == "")
+            return SLANG_E_NOT_AVAILABLE;
+    } while(1);
 
-    // We want the absolute path to the root
-    SLANG_RETURN_ON_FAIL(Path::getCanonical(rootRelPath, outExePath));
+    outExePath = std::move(rootRelPath);
     return SLANG_OK;
 }