|
|
|
@ -94,6 +94,7 @@ constexpr auto kJListArgs = "list_args";
|
|
|
|
|
constexpr auto kJSocVersion = "socVersion";
|
|
|
|
|
constexpr auto kSOC_VERSION = "SOC_VERSION";
|
|
|
|
|
constexpr auto kJIsDynamicShape = "is_dynamic_shape";
|
|
|
|
|
constexpr auto kJDynamicIndex = "dynamic_index";
|
|
|
|
|
|
|
|
|
|
bool IsNeedChangeDefaultFormat(const CNodePtr &cnode) {
|
|
|
|
|
MS_EXCEPTION_IF_NULL(cnode);
|
|
|
|
@ -137,6 +138,10 @@ bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptr<mindspor
|
|
|
|
|
op_info_json[kJAttrs] = attrs_json;
|
|
|
|
|
auto soc_version = TbeKernelJsonCreator::GetSocVersion();
|
|
|
|
|
op_info_json[kJSocVersion] = soc_version;
|
|
|
|
|
if (op_info_json[kJIsDynamicShape]) {
|
|
|
|
|
static int32_t dynamic_index = 0;
|
|
|
|
|
op_info_json[kJDynamicIndex] = dynamic_index++;
|
|
|
|
|
}
|
|
|
|
|
std::string json_str = op_info_json.dump();
|
|
|
|
|
size_t hash_id = std::hash<std::string>()(json_str);
|
|
|
|
|
auto context_ptr = MsContext::GetInstance();
|
|
|
|
|