[xpu] add missing dtypes for data_generate (#29)

PaddlePaddle · Oct 9, 2024 · a7a6cfc · a7a6cfc
1 parent 09452cb
commit a7a6cfc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/paddleapex/apex/utils/data_generate.py b/paddleapex/apex/utils/data_generate.py
@@ -43,6 +43,10 @@
     "paddle.float16",
     "paddle.half",
     "paddle.bfloat16",
+    "FLOAT64",
+    "FLOAT32",
+    "FLOAT16",
+    "BFLOAT16",
 ]
 REAL_TYPE_PADDLE = {
     "FP64": "paddle.float64",
@@ -54,6 +58,10 @@
     "INT16": "paddle.int16",
     "INT32": "paddle.int32",
     "INT64": "paddle.int64",
+    "FLOAT64": "paddle.float64",
+    "FLOAT32": "paddle.float32",
+    "FLOAT16": "paddle.float16",
+    "BFLOAT16": "paddle.bfloat16",
 }
 NUMPY_TYPE = [
     "numpy.int8",