From 612184759b8e5768afb35046a9196d72b497b751 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Mon, 16 Dec 2024 18:42:35 -0600 Subject: [PATCH] init target --- src/XLA.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/src/XLA.jl b/src/XLA.jl index 00420edb..b21999f9 100644 --- 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