From 6e5d8a455a452b0f7f21250afc17b3db48e3b892 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Tue, 28 Jan 2025 21:20:22 -0500 Subject: [PATCH] fix cpu --- src/enzyme_ad/jax/cpu.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/enzyme_ad/jax/cpu.cc b/src/enzyme_ad/jax/cpu.cc index 1b13d14c2..9e1e70c1e 100644 --- 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"); }