Update cpu.cc

EnzymeAD · Feb 2, 2025 · 2c6b907 · 2c6b907
1 parent 859aaf7
commit 2c6b907
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/enzyme_ad/jax/cpu.cc b/src/enzyme_ad/jax/cpu.cc
@@ -2,6 +2,7 @@
 
 struct CallInfo {
   void (*run)(const void **);
+  void *(*init)();
 };
 
 void forwarding_custom_call(void *out, const void **in, const CallInfo *opaque,