Expand checking

Author: rwy7

lib/Dialect/FIRRTL/Transforms/InferDomains.cpp

@@ -106,7 +106,6 @@ class CircuitDomainInfo {
   size_t getNumDomains() const { return domainTable.size(); }
   DomainOp getDomain(DomainTypeID id) const { return domainTable[id]; }
 
-
   DomainTypeID getDomainTypeID(StringAttr name) const {
     return typeIDTable.at(name);
   }
@@ -361,9 +360,9 @@ void solve(Term *lhs, Term *rhs) {
 // CheckModuleDomains
 //====--------------------------------------------------------------------------
 
-/// Check that a module has complete domain information.
-static LogicalResult checkModuleDomains(GlobalState &globals,
-                                        FModuleLike module) {
+/// Check that a module has complete domain information for its ports.
+static LogicalResult checkPortDomains(GlobalState &globals,
+                                      FModuleLike module) {
   auto numDomains = globals.circuitInfo.getNumDomains();
   auto domainInfo = module.getDomainInfoAttr();
   DenseMap<unsigned, unsigned> typeIDTable;
@@ -418,6 +417,19 @@ static LogicalResult checkModuleDomains(GlobalState &globals,
   return success();
 }
 
+static LogicalResult checkModuleDomains(GlobalState &globals,
+                                        FModuleOp module) {
+  if (failed(checkPortDomains(globals, module)))
+    return failure();
+
+  return success();
+}
+
+static LogicalResult checkModuleDomains(GlobalState &globals,
+                                        FExtModuleOp module) {
+  return checkPortDomains(globals, module);
+}
+
 //====--------------------------------------------------------------------------
 // InferModuleDomains: Primary workhorse for inferring domains on modules.
 //====--------------------------------------------------------------------------