ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the test

Browse Source
This commit is contained in:
Rostyslav Geyyer
2025-02-18 19:28:59 +00:00
parent af493e6fca
commit d90b50596b
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/data_type/test_mx_fp4.cpp
View File

@@ -338,9 +338,8 @@ __host__ __device__ float vec32_generator(ck::index_t i)
}
else
{
return type_convert<float>(ck::NumericLimits<f4_t>::Max()) -
vec16_generator(
i); // all positive values, then all negative values in descending order
return vec16_generator(
15 - (i % 16)); // all negative values, then all positive values in descending order
}
}
@@ -371,9 +370,9 @@ __global__ void test_mx_fp4x32_device_scaled_convert(float* p_test, uint64_t* p_
ck::static_for<0, N / 2, 1>{}([&](auto ii) {
p_test[i++] = type_convert<float>(
f4x32.AsType<f4x2_pk_t>()(ck::Number<ii>{}).template unpack<>(ck::Number<0>{}));
f4_t(f4x32.AsType<f4x2_pk_t>()(ck::Number<ii>{}).template unpack<>(ck::Number<0>{})));
p_test[i++] = type_convert<float>(
f4x32.AsType<f4x2_pk_t>()(ck::Number<ii>{}).template unpack<>(ck::Number<1>{}));
f4_t(f4x32.AsType<f4x2_pk_t>()(ck::Number<ii>{}).template unpack<>(ck::Number<1>{})));
});
}
@@ -424,7 +423,7 @@ __global__ void test_mx_fp4x32_device_scaled_convert_sr(float* p_test, uint64_t*
return;
}
auto scale2 = e8m0_bexp_t(8.0f);
auto scale2 = e8m0_bexp_t(2.0f);
f4x32_t f4x32{};
float32_t float32{};