init target

EnzymeAD · Dec 17, 2024 · 6121847 · 6121847
1 parent f6b2238
commit 6121847
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/XLA.jl b/src/XLA.jl
@@ -131,6 +131,7 @@ function __init__()
         end
     end
 
+    @ccall MLIR.API.mlir_c.RegisterCustomCallTarget("enzymexla_gpu"::Cstring, cglobal((:EnzymeGPUCustomCall, MLIR.API.mlir_c)), "CUDA")
     return nothing
 end