Check that no function->argumenttype mapping was passed where the fun…

…ction does not exist in the module

lib/Dialect/TTIR/Transforms/Utility.cpp

@@ -116,7 +116,11 @@ class TTIRPopulateArgumentTypes
                        << "\" argument count mismatch.\n";
           signalPassFailure();
         }
+        // Need to update/create the DictionaryAttr for each corresponding
+        // function argument.
         for (uint32_t i = 0; i < func.getNumArguments(); i++) {
+          // The current argument may already have attributes, so we need to add
+          // the argument type to that DictonaryAttr rather than overwrite it.
           std::optional<DictionaryAttr> currentArgAttrDict =
               func.getArgAttrDict(i)
                   ? std::make_optional(
@@ -125,6 +129,8 @@ class TTIRPopulateArgumentTypes
           std::vector<NamedAttribute> newArgAttrs;
           if (currentArgAttrDict.has_value()) {
             for (NamedAttribute currentArgAttr : currentArgAttrDict.value()) {
+              // If this argument already has an argumnet type, this pass wil
+              // overwrite it. Log a warning.
               if (currentArgAttr.getName() != "tt.argument_type") {
                 newArgAttrs.push_back(currentArgAttr);
               } else {
@@ -143,6 +149,16 @@ class TTIRPopulateArgumentTypes
         }
       }
     }
+
+    for (auto [funcName, argTypes] : argumentTypes) {
+      if (std::find(funcNames.begin(), funcNames.end(), funcName) ==
+          funcNames.end()) {
+        llvm::errs() << "Function: \"" << funcName
+                     << "\" was provided in the argument types map, however it "
+                        "was not found in module!\n";
+        signalPassFailure();
+      }
+    }
   }
 };