From 95939b522c39d85fa90916325e96b4e275892502 Mon Sep 17 00:00:00 2001
From: Feiyu Chan <chenfeiyu@baidu.com>
Date: Wed, 14 Apr 2021 11:00:30 +0800
Subject: [PATCH] add common dtypes as paddle's dtypes (#32012)

* add common dtypes as paddle's dtypes

* import paddle.fluid.core_avx.VarDesc.VarType as paddle.dtype
---
 python/paddle/__init__.py        | 13 +++++++++++
 python/paddle/framework/dtype.py | 40 ++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+)
 create mode 100644 python/paddle/framework/dtype.py

diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
index 17bf2d544f3..861839256a3 100755
--- a/python/paddle/__init__.py
+++ b/python/paddle/__init__.py
@@ -31,6 +31,19 @@ from .fluid.dygraph import monkey_patch_math_varbase
 monkey_patch_variable()
 monkey_patch_math_varbase()
 import paddle.framework
+from .framework.dtype import dtype as dtype
+from paddle.framework.dtype import uint8
+from paddle.framework.dtype import int8
+from paddle.framework.dtype import int16
+from paddle.framework.dtype import int32
+from paddle.framework.dtype import int64
+from paddle.framework.dtype import float16
+from paddle.framework.dtype import float32
+from paddle.framework.dtype import float64
+from paddle.framework.dtype import bfloat16
+from paddle.framework.dtype import bool
+from paddle.framework.dtype import complex64
+from paddle.framework.dtype import complex128
 from .framework import VarBase as Tensor
 import paddle.compat
 import paddle.distributed
diff --git a/python/paddle/framework/dtype.py b/python/paddle/framework/dtype.py
new file mode 100644
index 00000000000..3eeaa6e74ec
--- /dev/null
+++ b/python/paddle/framework/dtype.py
@@ -0,0 +1,40 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__all__ = [
+    "dtype", "uint8", "int8", "int16", "int32", "int64", "bfloat16", "float16",
+    "float32", "float64", "complex64", "complex128", "bool"
+]
+
+from ..fluid.core import VarDesc
+
+dtype = VarDesc.VarType
+dtype.__qualname__ = "dtype"
+dtype.__module__ = "paddle"
+
+uint8 = VarDesc.VarType.UINT8
+int8 = VarDesc.VarType.INT8
+int16 = VarDesc.VarType.INT16
+int32 = VarDesc.VarType.INT32
+int64 = VarDesc.VarType.INT64
+
+float32 = VarDesc.VarType.FP32
+float64 = VarDesc.VarType.FP64
+float16 = VarDesc.VarType.FP16
+bfloat16 = VarDesc.VarType.BF16
+
+complex64 = VarDesc.VarType.COMPLEX64
+complex128 = VarDesc.VarType.COMPLEX128
+
+bool = VarDesc.VarType.BOOL
-- 
GitLab