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

fix buffer size for output

Browse Source
This commit is contained in:
Matti Eskelinen
2026-02-06 09:33:33 +00:00
parent 3b31e42359
commit 683250e41b
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/ck_tile/sinkhorn/test_sinkhorn_impl.hpp
View File

@@ -38,9 +38,8 @@ class TestCkTileSinkHorn : public ::testing::Test
ck_tile::FillUniformDistribution<XDataType>{-5.f, 5.f}(h_x);
auto buffer_size = h_x.get_element_space_size_in_bytes();
ck_tile::DeviceMem d_x_mem(buffer_size);
ck_tile::DeviceMem d_y_mem(buffer_size);
ck_tile::DeviceMem d_x_mem(h_x.get_element_space_size_in_bytes());
ck_tile::DeviceMem d_y_mem(h_y.get_element_space_size_in_bytes());
ck_tile::SinkhornKnoppArgs args{static_cast<void*>(d_y_mem.GetDeviceBuffer()),
static_cast<void*>(d_x_mem.GetDeviceBuffer()),