diff --git a/tf_adapter/util/ge_plugin.cc b/tf_adapter/util/ge_plugin.cc index 41c8ad26eefbd04714b3559ad0deb4e06a853165..3f94a98f5841a3b40e4fe621a55471d83eca1e46 100644 --- a/tf_adapter/util/ge_plugin.cc +++ b/tf_adapter/util/ge_plugin.cc @@ -128,6 +128,7 @@ void SetOptionNameMap(json &option_name_map) { option_name_map.emplace("ge.executeTimes", "execute_times"); option_name_map.emplace(ge::OPTION_EXEC_DYNAMIC_EXECUTE_MODE, "dynamic_graph_execute_mode"); option_name_map.emplace(ge::OPTION_EXEC_DYNAMIC_INPUT, "dynamic_input"); + option_name_map.emplace(ge::AICORE_NUM, "aicore_num"); } } // namespace