Skip to content

[Bug Fix] Support casting lightweight float formats to complex types #75929

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
youge325 wants to merge 8 commits into from
Closed

[Bug Fix] Support casting lightweight float formats to complex types #75929

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
3327dcc
[Bug Fix] Support casting lightweight float formats to complex types
youge325 Oct 24, 2025
304a0a8
test: add test description comment to trigger coverage check
youge325 Oct 25, 2025
826c965
Add new test file for lightweight float to complex casting
youge325 Oct 25, 2025
6d01aa3
Remove @unittest.skipIf decorators to ensure tests run on CPU
youge325 Oct 26, 2025
e185958
Add C++ tests for lightweight float to complex conversions
youge325 Oct 26, 2025
8b26d82
[Bug Fix] Support casting lightweight float formats to complex types
youge325 Oct 24, 2025
88d3dd4
test: add test description comment to trigger coverage check
youge325 Oct 25, 2025
e016bf1
Add new test file for lightweight float to complex casting
youge325 Oct 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
72 changes: 72 additions & 0 deletions paddle/phi/core/framework/data_type_transform.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,78 @@ struct CastDataTypeFunctor {
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float8_e5m2,
::phi::dtype::complex<float>> {
HOSTDEVICE inline ::phi::dtype::complex<float> operator()(
::phi::dtype::float8_e5m2 in) const {
return ::phi::dtype::complex<float>(static_cast<float>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float8_e5m2,
::phi::dtype::complex<double>> {
HOSTDEVICE inline ::phi::dtype::complex<double> operator()(
::phi::dtype::float8_e5m2 in) const {
return ::phi::dtype::complex<double>(static_cast<double>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float8_e4m3fn,
::phi::dtype::complex<float>> {
HOSTDEVICE inline ::phi::dtype::complex<float> operator()(
::phi::dtype::float8_e4m3fn in) const {
return ::phi::dtype::complex<float>(static_cast<float>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float8_e4m3fn,
::phi::dtype::complex<double>> {
HOSTDEVICE inline ::phi::dtype::complex<double> operator()(
::phi::dtype::float8_e4m3fn in) const {
return ::phi::dtype::complex<double>(static_cast<double>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::bfloat16,
::phi::dtype::complex<float>> {
HOSTDEVICE inline ::phi::dtype::complex<float> operator()(
::phi::dtype::bfloat16 in) const {
return ::phi::dtype::complex<float>(static_cast<float>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::bfloat16,
::phi::dtype::complex<double>> {
HOSTDEVICE inline ::phi::dtype::complex<double> operator()(
::phi::dtype::bfloat16 in) const {
return ::phi::dtype::complex<double>(static_cast<double>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float16,
::phi::dtype::complex<float>> {
HOSTDEVICE inline ::phi::dtype::complex<float> operator()(
::phi::dtype::float16 in) const {
return ::phi::dtype::complex<float>(static_cast<float>(in));
}
};

template <>
struct CastDataTypeFunctor<::phi::dtype::float16,
::phi::dtype::complex<double>> {
HOSTDEVICE inline ::phi::dtype::complex<double> operator()(
::phi::dtype::float16 in) const {
return ::phi::dtype::complex<double>(static_cast<double>(in));
}
};

#if defined(PADDLE_WITH_XPU)

template <typename InType, typename OutType>
Expand Down
160 changes: 160 additions & 0 deletions test/cpp/fluid/framework/data_type_transform_test.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -395,4 +395,164 @@ TEST(DataTypeTransform, CPUTransform) {
EXPECT_EQ(ptr[i], static_cast<int32_t>(in_data_bool[i]));
}
}

// data type transform from lightweight float formats to complex types
{
auto kernel_float8_e5m2 = phi::KernelKey(
place, phi::DataLayout::ALL_LAYOUT, phi::DataType::FLOAT8_E5M2);
auto kernel_float8_e4m3fn = phi::KernelKey(
place, phi::DataLayout::ALL_LAYOUT, phi::DataType::FLOAT8_E4M3FN);
auto kernel_complex64 = phi::KernelKey(
place, phi::DataLayout::ALL_LAYOUT, phi::DataType::COMPLEX64);
auto kernel_complex128 = phi::KernelKey(
place, phi::DataLayout::ALL_LAYOUT, phi::DataType::COMPLEX128);

phi::DenseTensor in;
phi::DenseTensor out;
int data_number = 2 * 3;

// Test float16 to complex64
{
phi::dtype::float16* ptr = in.mutable_data<phi::dtype::float16>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float16>(i);
}

paddle::framework::TransDataType(kernel_fp16, kernel_complex64, in, &out);
phi::dtype::complex<float>* out_data =
out.data<phi::dtype::complex<float>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<float>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0f);
}
}

// Test float16 to complex128
{
phi::dtype::float16* ptr = in.mutable_data<phi::dtype::float16>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float16>(i);
}

paddle::framework::TransDataType(
kernel_fp16, kernel_complex128, in, &out);
phi::dtype::complex<double>* out_data =
out.data<phi::dtype::complex<double>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<double>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0);
}
}

// Test bfloat16 to complex64
{
phi::dtype::bfloat16* ptr = in.mutable_data<phi::dtype::bfloat16>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::bfloat16>(i);
}

paddle::framework::TransDataType(kernel_bf16, kernel_complex64, in, &out);
phi::dtype::complex<float>* out_data =
out.data<phi::dtype::complex<float>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<float>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0f);
}
}

// Test bfloat16 to complex128
{
phi::dtype::bfloat16* ptr = in.mutable_data<phi::dtype::bfloat16>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::bfloat16>(i);
}

paddle::framework::TransDataType(
kernel_bf16, kernel_complex128, in, &out);
phi::dtype::complex<double>* out_data =
out.data<phi::dtype::complex<double>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<double>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0);
}
}

// Test float8_e4m3fn to complex64
{
phi::dtype::float8_e4m3fn* ptr =
in.mutable_data<phi::dtype::float8_e4m3fn>(common::make_ddim({2, 3}),
place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float8_e4m3fn>(i);
}

paddle::framework::TransDataType(
kernel_float8_e4m3fn, kernel_complex64, in, &out);
phi::dtype::complex<float>* out_data =
out.data<phi::dtype::complex<float>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<float>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0f);
}
}

// Test float8_e4m3fn to complex128
{
phi::dtype::float8_e4m3fn* ptr =
in.mutable_data<phi::dtype::float8_e4m3fn>(common::make_ddim({2, 3}),
place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float8_e4m3fn>(i);
}

paddle::framework::TransDataType(
kernel_float8_e4m3fn, kernel_complex128, in, &out);
phi::dtype::complex<double>* out_data =
out.data<phi::dtype::complex<double>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<double>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0);
}
}

// Test float8_e5m2 to complex64
{
phi::dtype::float8_e5m2* ptr = in.mutable_data<phi::dtype::float8_e5m2>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float8_e5m2>(i);
}

paddle::framework::TransDataType(
kernel_float8_e5m2, kernel_complex64, in, &out);
phi::dtype::complex<float>* out_data =
out.data<phi::dtype::complex<float>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<float>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0f);
}
}

// Test float8_e5m2 to complex128
{
phi::dtype::float8_e5m2* ptr = in.mutable_data<phi::dtype::float8_e5m2>(
common::make_ddim({2, 3}), place);
for (int i = 0; i < data_number; ++i) {
ptr[i] = static_cast<phi::dtype::float8_e5m2>(i);
}

paddle::framework::TransDataType(
kernel_float8_e5m2, kernel_complex128, in, &out);
phi::dtype::complex<double>* out_data =
out.data<phi::dtype::complex<double>>();
for (int i = 0; i < data_number; ++i) {
EXPECT_EQ(out_data[i].real, static_cast<double>(ptr[i]));
EXPECT_EQ(out_data[i].imag, 0.0);
}
}
}
}
Loading
Loading