ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restrict the range of FillUniformDistributionIntegerValue for A and B to make tests pass

Browse Source
This commit is contained in:
Sami Aario
2026-01-26 14:59:28 +00:00
parent 298fd29fba
commit 1167950528
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/ck_tile/gemm/test_gemm_pipeline_util.hpp
View File

@@ -299,8 +299,8 @@ class TestCkTileGemmPipeline : public ::testing::Test
ck_tile::HostTensor<CDataType> c_m_n_dev_result(
ck_tile::host_tensor_descriptor(M, N, stride_C, is_row_major(CLayout{})));
ck_tile::FillUniformDistributionIntegerValue<ADataType>{-5, 5, 11939}(a_m_k);
ck_tile::FillUniformDistributionIntegerValue<BDataType>{-5, 5, 11940}(b_k_n);
ck_tile::FillUniformDistributionIntegerValue<ADataType>{-0.5, 0.5, 11939}(a_m_k);
ck_tile::FillUniformDistributionIntegerValue<BDataType>{-0.5, 0.5, 11940}(b_k_n);
ck_tile::DeviceMem a_m_k_dev_buf(a_m_k.get_element_space_size_in_bytes());
ck_tile::DeviceMem b_k_n_dev_buf(b_k_n.get_element_space_size_in_bytes());