support complex (#73886)

Author: cangtianhuang

paddle/phi/kernels/funcs/tensor_formatter.cc

@@ -115,6 +115,10 @@ std::string TensorFormatter::Format(const phi::DenseTensor& print_tensor,
  FormatData<phi::dtype::float8_e4m3fn>(print_tensor, log_stream);
  } else if (dtype == phi::DataType::FLOAT8_E5M2) {
  FormatData<phi::dtype::float8_e5m2>(print_tensor, log_stream);
+ } else if (dtype == phi::DataType::COMPLEX64) {
+ FormatData<phi::dtype::complex<float>>(print_tensor, log_stream);
+ } else if (dtype == phi::DataType::COMPLEX128) {
+ FormatData<phi::dtype::complex<double>>(print_tensor, log_stream);
  } else {
  log_stream << " - data: unprintable type: " << dtype << std::endl;
  }
@@ -143,9 +147,20 @@ void TensorFormatter::FormatData(const phi::DenseTensor& print_tensor,
 
  log_stream << " - data: [";
  if (print_size > 0) {
- log_stream << data[0];
+ auto print_element = [&log_stream](const auto& elem) {
+ if constexpr (std::is_same_v<T, phi::dtype::complex<float>> ||
+ std::is_same_v<T, phi::dtype::complex<double>>) {
+ log_stream << static_cast<float>(elem.real) << "+"
+ << static_cast<float>(elem.imag) << "j";
+ } else {
+ log_stream << static_cast<float>(elem);
+ }
+ };
+
+ print_element(data[0]);
  for (int64_t i = 1; i < print_size; ++i) {
- log_stream << " " << static_cast<float>(data[i]);
+ log_stream << " ";
+ print_element(data[i]);
  }
  }
  log_stream << "]" << std::endl;
@@ -165,6 +180,10 @@ template void TensorFormatter::FormatData<phi::dtype::float16>(
  const phi::DenseTensor& print_tensor, std::stringstream& log_stream);
 template void TensorFormatter::FormatData<phi::dtype::bfloat16>(
  const phi::DenseTensor& print_tensor, std::stringstream& log_stream);
+template void TensorFormatter::FormatData<phi::dtype::complex<float>>(
+ const phi::DenseTensor& print_tensor, std::stringstream& log_stream);
+template void TensorFormatter::FormatData<phi::dtype::complex<double>>(
+ const phi::DenseTensor& print_tensor, std::stringstream& log_stream);
 
 } // namespace funcs
 } // namespace paddle