From d300969d544eca5e6dfeb7ac38f60dccad1902f3 Mon Sep 17 00:00:00 2001 From: sbrantq Date: Thu, 30 Oct 2025 23:20:35 -0500 Subject: [PATCH 1/2] hmc jll changes --- deps/ReactantExtra/API.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp index cdede7c364..41f4f7f1dc 100644 --- a/deps/ReactantExtra/API.cpp +++ b/deps/ReactantExtra/API.cpp @@ -388,6 +388,18 @@ enzymeSymbolAttrGet(MlirContext ctx, uint64_t symbol) { return wrap(attr); } +extern "C" MLIR_CAPI_EXPORTED MlirAttribute +enzymeRngDistributionAttrGet(MlirContext ctx, int32_t val) { + return wrap(mlir::enzyme::RngDistributionAttr::get( + unwrap(ctx), (mlir::enzyme::RngDistribution)val)); +} + +extern "C" MLIR_CAPI_EXPORTED MlirAttribute +enzymeMCMCAlgorithmAttrGet(MlirContext ctx, int32_t val) { + return wrap(mlir::enzyme::MCMCAlgorithmAttr::get( + unwrap(ctx), (mlir::enzyme::MCMCAlgorithm)val)); +} + // Create profiler session and start profiling REACTANT_ABI tsl::ProfilerSession * CreateProfilerSession(uint32_t device_tracer_level, From 68c4368c5e13e9de870b84f689d38e11f765d5be Mon Sep 17 00:00:00 2001 From: sbrantq Date: Fri, 31 Oct 2025 00:11:05 -0500 Subject: [PATCH 2/2] fix --- deps/ReactantExtra/API.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp index 41f4f7f1dc..18636d89d0 100644 --- a/deps/ReactantExtra/API.cpp +++ b/deps/ReactantExtra/API.cpp @@ -373,28 +373,28 @@ enzymeActivityAttrGet(MlirContext ctx, int32_t val) { (mlir::enzyme::Activity)val)); } -extern "C" MLIR_CAPI_EXPORTED MlirType enzymeTraceTypeGet(MlirContext ctx) { +REACTANT_ABI MLIR_CAPI_EXPORTED MlirType enzymeTraceTypeGet(MlirContext ctx) { return wrap(mlir::enzyme::TraceType::get(unwrap(ctx))); } -extern "C" MLIR_CAPI_EXPORTED MlirType +REACTANT_ABI MLIR_CAPI_EXPORTED MlirType enzymeConstraintTypeGet(MlirContext ctx) { return wrap(mlir::enzyme::ConstraintType::get(unwrap(ctx))); } -extern "C" MLIR_CAPI_EXPORTED MlirAttribute +REACTANT_ABI MLIR_CAPI_EXPORTED MlirAttribute enzymeSymbolAttrGet(MlirContext ctx, uint64_t symbol) { mlir::Attribute attr = mlir::enzyme::SymbolAttr::get(unwrap(ctx), symbol); return wrap(attr); } -extern "C" MLIR_CAPI_EXPORTED MlirAttribute +REACTANT_ABI MLIR_CAPI_EXPORTED MlirAttribute enzymeRngDistributionAttrGet(MlirContext ctx, int32_t val) { return wrap(mlir::enzyme::RngDistributionAttr::get( unwrap(ctx), (mlir::enzyme::RngDistribution)val)); } -extern "C" MLIR_CAPI_EXPORTED MlirAttribute +REACTANT_ABI MLIR_CAPI_EXPORTED MlirAttribute enzymeMCMCAlgorithmAttrGet(MlirContext ctx, int32_t val) { return wrap(mlir::enzyme::MCMCAlgorithmAttr::get( unwrap(ctx), (mlir::enzyme::MCMCAlgorithm)val));