From 59c68c159171dadb9d9b858efc053f6e8b430304 Mon Sep 17 00:00:00 2001
From: tianhaodongbd <tianhaodong@baidu.com>
Date: Tue, 8 Oct 2024 14:12:05 +0800
Subject: [PATCH] [XPU] add missing DataType

---
 paddleapex/api_tracer/api_info.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddleapex/api_tracer/api_info.py b/paddleapex/api_tracer/api_info.py
index df293fd..672b0d6 100644
--- a/paddleapex/api_tracer/api_info.py
+++ b/paddleapex/api_tracer/api_info.py
@@ -27,6 +27,10 @@
     "INT16": "paddle.int16",
     "INT32": "paddle.int32",
     "INT64": "paddle.int64",
+    "FLOAT64": "paddle.float64",
+    "FLOAT32": "paddle.float32",
+    "FLOAT16": "paddle.float16",
+    "BFLOAT16": "paddle.bfloat16",
 }