tf_function_not_found

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 15, 2024 · d12344b · d12344b
1 parent 59952fa
commit d12344b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/source/api_cc/src/DeepPotJAX.cc b/source/api_cc/src/DeepPotJAX.cc
@@ -64,7 +64,11 @@ inline TF_DataType get_data_tensor_type(const std::vector<int64_t>& data) {
   return TF_INT64;
 }
 
-struct tf_function_not_found : public deepmd::deepmd_exception {};
+struct tf_function_not_found : public deepmd::deepmd_exception {
+ public:
+  tf_function_not_found() : deepmd_exception() {};
+  tf_function_not_found(const std::string& msg) : deepmd_exception(msg) {};
+};
 
 inline TFE_Op* get_func_op(TFE_Context* ctx,
                            const std::string func_name,