fix xpu_kp bugs (#54377)

5bec661f · YuanRisheng · GitHub · 0942f77e · 5bec661f
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/phi/core/utils/type_info.cc paddle/phi/core/utils/type_info.cc +3 -0

未找到文件。
--- a/paddle/phi/core/utils/type_info.cc
+++ b/paddle/phi/core/utils/type_info.cc
@@ -55,6 +55,9 @@ template class TypeInfoTraits<phi::DeviceContext, CustomContext>;
 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) || \
    defined(PADDLE_WITH_XPU_KP)
 template class TypeInfoTraits<phi::DeviceContext, GPUContext>;
+#endif
+
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 template class TypeInfoTraits<phi::DeviceContext, GPUPinnedContext>;
 #endif