fix compile bugs when run kunlun

Author: YuanRisheng

paddle/pten/kernels/xpu/manipulation.cc

@@ -55,13 +55,16 @@ void FlattenWithXShape(const XPUDeviceContext& dev_ctx,
 // TODO(chenweihang): replace by better impl
 PT_REGISTER_MODULE(ManipulationXPU);
 
+using float16 = paddle::platform::float16;
+
 // TODO(yuanrisheng): "flatten_contiguous_range" is compatible with old kernel
 // architecture, kernel_name should be "flatten".
 PT_REGISTER_KERNEL("flatten_contiguous_range",
  XPU,
  ANY,
  pten::Flatten,
  float,
+ float16,
  double,
  uint8_t,
  int8_t,
@@ -73,6 +76,7 @@ PT_REGISTER_KERNEL("flatten_contiguous_range.mid",
  ANY,
  pten::FlattenWithXShape,
  float,
+ float16,
  double,
  uint8_t,
  int8_t,