export map symbol (#1313)

* export map symbol

* fmt

* fix

Author: wsmoses

src/enzyme_ad/jax/BUILD

@@ -603,6 +603,7 @@ cc_library(
         "@llvm-project//mlir:VectorToLLVM",
         "@llvm-project//mlir:VectorToSCF",
         "@llvm-project//mlir:ViewLikeInterface",
+        "@llvm-project//mlir:CAPIIR",
         "@shardy//shardy/dialect/sdy/ir:dialect",
         "@shardy//shardy/dialect/sdy/transforms/propagation:op_sharding_rule_builder",
         "@stablehlo//:base",

src/enzyme_ad/jax/Passes/LowerJIT.cpp

@@ -80,6 +80,8 @@
 
 #include "mlir/Target/LLVMIR/Export.h"
 
+#include "mlir-c/Support.h"
+
 #define DEBUG_TYPE "lower-jit"
 
 namespace mlir {
@@ -274,7 +276,8 @@ bool initJIT() {
   return true;
 }
 
-extern "C" void EnzymeJaXMapSymbol(const char *name, void *symbol) {
+extern "C" MLIR_CAPI_EXPORTED void EnzymeJaXMapSymbol(const char *name,
+                                                      void *symbol) {
   initJIT();
   MappedSymbols[JIT->mangleAndIntern(name)] = llvm::orc::ExecutorSymbolDef(
       llvm::orc::ExecutorAddr::fromPtr(symbol), llvm::JITSymbolFlags());