probprog attr mlir api

Author: sbrantq

deps/ReactantExtra/API.cpp

@@ -364,6 +364,16 @@ enzymeActivityAttrGet(MlirContext ctx, int32_t val) {
                                               (mlir::enzyme::Activity)val));
 }
 
+extern "C" MLIR_CAPI_EXPORTED MlirType enzymeTraceTypeGet(MlirContext ctx) {
+  return wrap(mlir::enzyme::TraceType::get(unwrap(ctx)));
+}
+
+extern "C" 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 enzymeConstraintAttrGet(
     MlirContext ctx, uint64_t symbol, MlirAttribute values) {
   mlir::Attribute vals = unwrap(values);