fix memory optimization type (#19781)

test=develop
5 years ago · 110be57c1b
parent 5fbf03d6b4
commit 110be57c1b
1 changed files with 10 additions and 0 deletions
--- a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc
+++ b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc
@ -101,6 +101,16 @@ int DataTypeToSpace(framework::proto::VarType_Type type) {
      return sizeof(int32_t);
    case framework::proto::VarType_Type_INT64:
      return sizeof(int64_t);
+    case framework::proto::VarType_Type_INT16:
+      return sizeof(int16_t);
+    case framework::proto::VarType_Type_FP16:
+      return sizeof(int16_t);
+    case framework::proto::VarType_Type_FP64:
+      return sizeof(double);
+    case framework::proto::VarType_Type_UINT8:
+      return sizeof(unsigned char);
+    case framework::proto::VarType_Type_INT8:
+      return sizeof(int8_t);
    default:
      PADDLE_THROW("Unknown data type");
  }