diff options
| -rw-r--r-- | source/compiler-core/slang-nvrtc-compiler.cpp | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/source/compiler-core/slang-nvrtc-compiler.cpp b/source/compiler-core/slang-nvrtc-compiler.cpp index 162e91353..950a1fc6e 100644 --- a/source/compiler-core/slang-nvrtc-compiler.cpp +++ b/source/compiler-core/slang-nvrtc-compiler.cpp @@ -127,7 +127,7 @@ protected: nvrtcProgram m_program; }; - SlangResult _findCUDAIncludePath(String& outIncludePath); + SlangResult _findCUDAIncludePath(String& outPath); SlangResult _getCUDAIncludePath(String& outIncludePath); SlangResult _findOptixIncludePath(String& outIncludePath); @@ -706,10 +706,13 @@ SlangResult NVRTCDownstreamCompiler::_findCUDAIncludePath(String& outPath) } #if SLANG_LINUX_FAMILY - // Try /usr/include - { - String includePath = "/usr/include"; + List<String> candidatePaths; + candidatePaths.add("/usr/local/include"); + candidatePaths.add("/usr/local/cuda/include"); + candidatePaths.add("/usr/include"); + for (const String& includePath : candidatePaths) + { if (File::exists(Path::combine(includePath, g_fp16HeaderName))) { outPath = includePath; @@ -717,7 +720,6 @@ SlangResult NVRTCDownstreamCompiler::_findCUDAIncludePath(String& outPath) } } #endif - return SLANG_E_NOT_FOUND; } |