[FIRRTL][CAPI] add layer for firrtlTypeGetRef

llvm · Jan 17, 2025 · ff1fc2e · ff1fc2e
1 parent 375a8e0
commit ff1fc2e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/include/circt-c/Dialect/FIRRTL.h b/include/circt-c/Dialect/FIRRTL.h
@@ -208,7 +208,8 @@ firrtlTypeGetBundleFieldIndex(MlirType type, MlirStringRef fieldName);
 MLIR_CAPI_EXPORTED bool firrtlTypeIsARef(MlirType type);
 
 /// Creates a ref type.
-MLIR_CAPI_EXPORTED MlirType firrtlTypeGetRef(MlirType target, bool forceable);
+MLIR_CAPI_EXPORTED MlirType firrtlTypeGetRef(MlirType target, bool forceable,
+                                             MlirAttribute layer);
 
 /// Checks if this type is an anyref type.
 MLIR_CAPI_EXPORTED bool firrtlTypeIsAAnyRef(MlirType type);

diff --git a/lib/CAPI/Dialect/FIRRTL.cpp b/lib/CAPI/Dialect/FIRRTL.cpp
@@ -185,11 +185,12 @@ unsigned firrtlTypeGetBundleFieldIndex(MlirType type, MlirStringRef fieldName) {
 
 bool firrtlTypeIsARef(MlirType type) { return isa<RefType>(unwrap(type)); }
 
-MlirType firrtlTypeGetRef(MlirType target, bool forceable) {
+MlirType firrtlTypeGetRef(MlirType target, bool forceable, MlirAttribute layer) {
   auto baseType = dyn_cast<FIRRTLBaseType>(unwrap(target));
   assert(baseType && "target must be base type");
+  auto layerAttr = dyn_cast<SymbolRefAttr>(unwrap(layer));
 
-  return wrap(RefType::get(baseType, forceable));
+  return wrap(RefType::get(baseType, forceable, layerAttr));
 }
 
 bool firrtlTypeIsAAnyRef(MlirType type) {