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

ck-builder: tensor input/output reflection (#3536)

Browse Source 
This adds some utilities to automatically generate UniqueInputs,
UniqueOutputs, alloc_inputs, alloc_outputs, and validate, based
on a Inputs::reflect() and Outputs::reflect().
This commit is contained in:
Robin Voetter
2026-01-12 09:45:53 +01:00
committed by GitHub
parent 32408c8bc0
commit b352a68606
8 changed files with 299 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
experimental/builder/test/unit_device_buffer.cpp
View File

@@ -88,3 +88,11 @@ TEST(DeviceBuffer, AllocTensorBuffer)
EXPECT_THAT(hipMemset(buffer.get(), 0xFF, descriptor.get_element_space_size_in_bytes()),
HipSuccess());
}
TEST(DeviceBuffer, AlignForward)
{
EXPECT_THAT(ckt::align_fwd(24, 8), Eq(24));
EXPECT_THAT(ckt::align_fwd(25, 8), Eq(32));
EXPECT_THAT(ckt::align_fwd(0xd7c563, 0x1000), Eq(0xd7d000));
EXPECT_THAT(ckt::align_fwd(19561, 23), Eq(19573));
}