diff --git a/tf_adapter/util/ge_plugin.cc b/tf_adapter/util/ge_plugin.cc index dbad404f2a8a54141e184f9090b263c41e0e9287..340951a72844bd3b8f25eb7a654902884f9da1c8 100644 --- a/tf_adapter/util/ge_plugin.cc +++ b/tf_adapter/util/ge_plugin.cc @@ -269,7 +269,7 @@ uint64_t GePlugin::GetFusionTensorSize() { } std::string temp_fusion_tensor_size(env_fusion_tensor_size); std::istringstream string_stream; - if (!temp_fusion_tensor_size.empty()) { + if ((!temp_fusion_tensor_size.empty()) && (temp_fusion_tensor_size.length() < ADAPTER_ENV_MAX_LENTH)) { string_stream.str(temp_fusion_tensor_size); } uint64_t fusion_tensor_size = 0;