Skip to content

Commit cbb50be

Browse files
authored
[Compiler] Set toolkit to XLA.CUDA_DATA_DIR (#1281)
1 parent dc08630 commit cbb50be

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

src/Compiler.jl

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1190,10 +1190,7 @@ function compile_mlir!(
11901190

11911191
optimize isa Bool && (optimize = ifelse(optimize, :all, :none))
11921192

1193-
toolkit = ""
1194-
if isdefined(Reactant_jll, :ptxas_path)
1195-
toolkit = Reactant_jll.ptxas_path[1:(end - length("/bin/ptxas"))]
1196-
end
1193+
toolkit = XLA.CUDA_DATA_DIR[]
11971194

11981195
if backend == "cpu" || backend == "tpu"
11991196
kern = "lower-kernel{backend=cpu},canonicalize"

0 commit comments

Comments
 (0)