Merge pull request #29 from gomlx/fixed-nvidia-cuda-path

janpfeifer · web-flow · commit 66f9f35325ef · 2026-03-18T19:11:37.000+01:00
- Fixed CUDA library paths -- it was hardcoded to a test directory.
diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md
@@ -1,3 +1,7 @@
+# v0.2.1: CUDA plugin fixes
+
+- Fixed CUDA library paths -- it was hardcoded to a test directory.
+
 # v0.2.0: Dynamic Shapes (thx @ajroetker); added `Call()`; Quantized shapes;
 
 - `go.mod`: updated to Go 1.25: needed for `synctest`.
diff --git a/pkg/pjrt/cuda_linux.go b/pkg/pjrt/cuda_linux.go
@@ -94,8 +94,7 @@ func cudaPluginCheckDrivers(name string) {
 			plugin.Path(), nvidiaPath, plugin.Path(), nvidiaPath)
 		return
 	}
-	// cudaSetCUDADir(nvidiaPath)
-	cudaSetCUDADir("/home/janpf/.local/lib/go-xla/nvidia")
+	cudaSetCUDADir(nvidiaPath)
 }
 
 // cudaNVidiaPath returns the path to the nvidia libraries. If they are not found in the same directory as the plugin.

Original file line number	Diff line number	Diff line change
`@@ -94,8 +94,7 @@ func cudaPluginCheckDrivers(name string) {`
`94`	`94`	`plugin.Path(), nvidiaPath, plugin.Path(), nvidiaPath)`
`95`	`95`	`return`
`96`	`96`	`}`
`97`		`- // cudaSetCUDADir(nvidiaPath)`
`98`		`- cudaSetCUDADir("/home/janpf/.local/lib/go-xla/nvidia")`
	`97`	`+ cudaSetCUDADir(nvidiaPath)`
`99`	`98`	`}`
`100`	`99`
`101`	`100`	`// cudaNVidiaPath returns the path to the nvidia libraries. If they are not found in the same directory as the plugin.`