Browse Source

abandon using EXPERIMENTAL_DYNAMIC_PARTITION

tags/v1.2.0
lichun 3 years ago
parent
commit
0a719fc4b7
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      ge/graph/partition/dynamic_shape_partition.cc

+ 10
- 0
ge/graph/partition/dynamic_shape_partition.cc View File

@@ -56,6 +56,11 @@ static bool IsContainResourceOp(const ComputeGraphPtr &root_graph) {
if (type == DT_STRING || type == DT_RESOURCE || type == DT_STRING_REF) { if (type == DT_STRING || type == DT_RESOURCE || type == DT_STRING_REF) {
if (std::getenv("EXPERIMENTAL_DYNAMIC_PARTITION") == nullptr) { if (std::getenv("EXPERIMENTAL_DYNAMIC_PARTITION") == nullptr) {
return false; return false;
} else {
GELOGE(FAILED, "In dynamic shape scene, model contains data type:"
"DT_STRING/DT_RESOURCE/DT_STRING_REF may not be supported well "
"temporarily, please retry with \"unset EXPERIMENTAL_DYNAMIC_PARTITION\".");
break;
} }
} }
} }
@@ -64,6 +69,11 @@ static bool IsContainResourceOp(const ComputeGraphPtr &root_graph) {
if (type == DT_STRING || type == DT_RESOURCE || type == DT_STRING_REF) { if (type == DT_STRING || type == DT_RESOURCE || type == DT_STRING_REF) {
if (std::getenv("EXPERIMENTAL_DYNAMIC_PARTITION") == nullptr) { if (std::getenv("EXPERIMENTAL_DYNAMIC_PARTITION") == nullptr) {
return false; return false;
} else {
GELOGE(FAILED, "In dynamic shape scene, model contains data type:"
"DT_STRING/DT_RESOURCE/DT_STRING_REF may not be supported well "
"temporarily, please retry with \"unset EXPERIMENTAL_DYNAMIC_PARTITION\".");
break;
} }
} }
} }


Loading…
Cancel
Save