[ET-VK][EZ] Throw in VK_GET_OP_FN if op is not found

jorgep31415 · jorgep31415 · commit 9b2019b5fc03 · 2024-04-12T15:38:43.000-07:00
Make yipjustin happy. Forgot this safeguard when I originally wrote the `OperatorRegistry` class. Differential Revision: [D56085588](https://our.internmc.facebook.com/intern/diff/D56085588/) ghstack-source-id: 222384998 Pull Request resolved: #3028
diff --git a/backends/vulkan/runtime/graph/ops/OperatorRegistry.cpp b/backends/vulkan/runtime/graph/ops/OperatorRegistry.cpp
@@ -16,7 +16,9 @@ bool OperatorRegistry::has_op(const std::string& name) {
 
 OperatorRegistry::OpFunction& OperatorRegistry::get_op_fn(
     const std::string& name) {
-  return table_.find(name)->second;
+  const auto it = table_.find(name);
+  VK_CHECK_COND(it != table_.end(), "Could not find operator with name ", name);
+  return it->second;
 }
 
 void OperatorRegistry::register_op(const std::string& name, OpFunction& fn) {

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,9 @@ bool OperatorRegistry::has_op(const std::string& name) {`
`16`	`16`
`17`	`17`	`OperatorRegistry::OpFunction& OperatorRegistry::get_op_fn(`
`18`	`18`	`const std::string& name) {`
`19`		`- return table_.find(name)->second;`
	`19`	`+ const auto it = table_.find(name);`
	`20`	`+ VK_CHECK_COND(it != table_.end(), "Could not find operator with name ", name);`
	`21`	`+ return it->second;`
`20`	`22`	`}`
`21`	`23`
`22`	`24`	`void OperatorRegistry::register_op(const std::string& name, OpFunction& fn) {`