|
14 | 14 |
|
15 | 15 | #include "gtest/gtest.h" |
16 | 16 | #include "paddle/fluid/framework/op_registry.h" |
17 | | -#include "paddle/fluid/platform/float16.h" |
18 | 17 | #include "paddle/phi/backends/xpu/xpu_info.h" // Include XPU initialization headers |
| 18 | +#include "paddle/phi/common/float16.h" |
19 | 19 | #include "paddle/phi/core/kernel_registry.h" |
20 | 20 |
|
21 | 21 | template <typename Place, typename T> |
@@ -193,20 +193,20 @@ TEST(SaveLoadCombineOp, XPU) { |
193 | 193 | EXPECT_EQ(r, 0); |
194 | 194 |
|
195 | 195 | r = SaveLoadCombineOpTest<phi::XPUPlace, |
196 | | - paddle::platform::float16, |
197 | | - paddle::platform::float16>(xpu_place); |
| 196 | + phi::dtype::float16, |
| 197 | + phi::dtype::float16>(xpu_place); |
198 | 198 | EXPECT_EQ(r, 0); |
199 | 199 | r = SaveLoadCombineOpTest<phi::CPUPlace, |
200 | | - paddle::platform::float16, |
201 | | - paddle::platform::float16>(cpu_place); |
| 200 | + phi::dtype::float16, |
| 201 | + phi::dtype::float16>(cpu_place); |
202 | 202 | EXPECT_EQ(r, 0); |
203 | 203 |
|
204 | 204 | r = SaveLoadCombineOpTest<phi::XPUPlace, |
205 | | - paddle::platform::bfloat16, |
206 | | - paddle::platform::bfloat16>(xpu_place); |
| 205 | + phi::dtype::bfloat16, |
| 206 | + phi::dtype::bfloat16>(xpu_place); |
207 | 207 | EXPECT_EQ(r, 0); |
208 | 208 | r = SaveLoadCombineOpTest<phi::CPUPlace, |
209 | | - paddle::platform::bfloat16, |
210 | | - paddle::platform::bfloat16>(cpu_place); |
| 209 | + phi::dtype::bfloat16, |
| 210 | + phi::dtype::bfloat16>(cpu_place); |
211 | 211 | EXPECT_EQ(r, 0); |
212 | 212 | } |
0 commit comments