fix cpu

EnzymeAD · Jan 29, 2025 · 6e5d8a4 · 6e5d8a4
1 parent 0805045
commit 6e5d8a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/enzyme_ad/jax/cpu.cc b/src/enzyme_ad/jax/cpu.cc
@@ -11,5 +11,5 @@ void forwarding_custom_call(void *out, const void **in, const CallInfo *opaque,
 
 extern "C" void RegisterEnzymeXLACPUHandler() {
   CustomCallTargetRegistry::Global()->Register("enzymexla_compile_cpu",
-                                               forwarding_custom_call, "Host");
+                                               &forwarding_custom_call, "Host");
 }