Skip to content

Commit 32bb8b2

Browse files
committed
fix: detect and solve more CUDA compilation issues
1 parent 335a85a commit 32bb8b2

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/bindings/utils/compileLLamaCpp.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,10 @@ export async function compileLlamaCpp(buildOptions: BuildOptions, compileOptions
253253
err.combinedStd.toLowerCase().includes(
254254
'variable "CUDACXX" or the CMake cache entry CMAKE_CUDA_COMPILER to the full'.toLowerCase()
255255
)
256+
) || (
257+
err.combinedStd.toLowerCase().includes("The CUDA compiler".toLowerCase()) &&
258+
err.combinedStd.toLowerCase().includes("is not able to compile a simple test program".toLowerCase()) &&
259+
err.combinedStd.toLowerCase().includes("nvcc fatal".toLowerCase())
256260
)
257261
)) {
258262
for (const {nvccPath, cudaHomePath} of await getCudaNvccPaths()) {

0 commit comments

Comments
 (0)