修改获取Param的方法

zjhellofss · Apr 10, 2024 · fb715cf · fb715cf
1 parent 3722888
commit fb715cf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/include/runtime/runtime_op.hpp b/include/runtime/runtime_op.hpp
@@ -96,7 +96,13 @@ struct RuntimeOperatorBase {
 
 template <typename T>
 bool RuntimeOperatorBase<T>::has_attribute(const std::string& attr_name) {
-  return false;
+  if (this->attribute.empty()) {
+    return false;
+  }
+  if (this->attribute.find(attr_name) == this->attribute.end()) {
+    return false;
+  }
+  return true;
 }
 
 template <typename T>