include/circt-c/Dialect/RTG.h

@@ -20,6 +20,7 @@ extern "C" {
 //===----------------------------------------------------------------------===//
 
 MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(RTG, rtg);
+MLIR_CAPI_EXPORTED void registerRTGPasses(void);
 
 //===----------------------------------------------------------------------===//
 // Type API.
@@ -35,8 +36,7 @@ MLIR_CAPI_EXPORTED MlirType rtgSequenceTypeGet(MlirContext ctxt);
 MLIR_CAPI_EXPORTED bool rtgTypeIsASet(MlirType type);
 
 /// Creates an RTG set type in the context.
-MLIR_CAPI_EXPORTED MlirType rtgSetTypeGet(MlirContext ctxt,
-                                          MlirType elementType);
+MLIR_CAPI_EXPORTED MlirType rtgSetTypeGet(MlirType elementType);
 
 /// If the type is an RTG dict.
 MLIR_CAPI_EXPORTED bool rtgTypeIsADict(MlirType type);

include/circt-c/RtgTool.h

@@ -62,6 +62,9 @@ MLIR_CAPI_EXPORTED void circtRtgToolOptionsSetUnsupportedInstructions(
     CirctRtgToolOptions options, unsigned numInstr,
     const char **unsupportedInstructions);
 
+MLIR_CAPI_EXPORTED void circtRtgToolOptionsAddUnsupportedInstruction(
+    CirctRtgToolOptions options, const char *unsupportedInstruction);
+
 MLIR_CAPI_EXPORTED void
 circtRtgToolOptionsSetUnsupportedInstructionsFile(CirctRtgToolOptions options,
                                                   const char *filename);

lib/CAPI/Dialect/CMakeLists.txt

@@ -127,6 +127,7 @@ add_circt_public_c_api_library(CIRCTCAPIRTG
   LINK_LIBS PUBLIC
   MLIRCAPIIR
   CIRCTRTGDialect
+  CIRCTRTGTransforms
 )
 
 if(CIRCT_INCLUDE_TESTS)

lib/CAPI/Dialect/RTG.cpp

@@ -10,6 +10,7 @@
 #include "circt/Dialect/RTG/IR/RTGDialect.h"
 #include "circt/Dialect/RTG/IR/RTGTypes.h"
 
+#include "circt/Dialect/RTG/Transforms/RTGPasses.h"
 #include "mlir/CAPI/Registration.h"
 
 using namespace circt;
@@ -20,6 +21,7 @@ using namespace circt::rtg;
 //===----------------------------------------------------------------------===//
 
 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(RTG, rtg, RTGDialect)
+void registerRTGPasses() { registerPasses(); }
 
 //===----------------------------------------------------------------------===//
 // Type API.
@@ -41,8 +43,9 @@ MlirType rtgSequenceTypeGet(MlirContext ctxt) {
 
 bool rtgTypeIsASet(MlirType type) { return isa<SetType>(unwrap(type)); }
 
-MlirType rtgSetTypeGet(MlirContext ctxt, MlirType elementType) {
-  return wrap(SetType::get(unwrap(ctxt), unwrap(elementType)));
+MlirType rtgSetTypeGet(MlirType elementType) {
+  auto ty = unwrap(elementType);
+  return wrap(SetType::get(ty.getContext(), ty));
 }
 
 // DictType

lib/CAPI/RtgTool/RtgTool.cpp

@@ -79,6 +79,12 @@ void circtRtgToolOptionsSetUnsupportedInstructions(
   unwrap(options)->setUnsupportedInstructions(std::move(instr));
 }
 
+void circtRtgToolOptionsAddUnsupportedInstruction(
+    CirctRtgToolOptions options, const char *unsupportedInstruction) {
+  unwrap(options)->addUnsupportedInstruction(
+      std::string(unsupportedInstruction));
+}
+
 void circtRtgToolOptionsSetUnsupportedInstructionsFile(
     CirctRtgToolOptions options, const char *filename) {
   unwrap(options)->setUnsupportedInstructionsFile(std::string(filename));

test/CAPI/rtg.c

@@ -26,7 +26,7 @@ static void testSequenceType(MlirContext ctx) {
 
 static void testSetType(MlirContext ctx) {
   MlirType elTy = mlirIntegerTypeGet(ctx, 32);
-  MlirType setTy = rtgSetTypeGet(ctx, elTy);
+  MlirType setTy = rtgSetTypeGet(elTy);
 
   // CHECK: is_set
   fprintf(stderr, rtgTypeIsASet(setTy) ? "is_set\n" : "isnot_set\n");