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

Add 8D tensor descriptor tests

Browse Source 
Test high-dimensional tensors to verify no integer overflow in
element space size calculations. Includes:
- 8D packed test with prime dimensions (180180 elements)
- 8D permuted layout test with non-monotonous strides
  (memory order differs from logical order)
This commit is contained in:
Max Podkorytov
2026-01-20 10:43:49 -06:00
parent 1a5d956be7
commit e1f2d45ef4
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
test/util/unit_tensor_descriptor_helper.cpp
View File

@@ -160,3 +160,29 @@ TEST(MakeNaiveTensorDescriptorAligned, Align8)
constexpr auto desc = make_naive_tensor_descriptor_aligned(lengths, Number<8>{});
EXPECT_EQ(desc.GetElementSpaceSize(), 119);
}
// Test high-dimensional tensors (8D) to verify no integer overflow
TEST(MakeNaiveTensorDescriptorPacked, Simple8D)
{
// 8D packed tensor with small prime dimensions: 2x3x5x7x11x13x2x3
// element_space_size = 2*3*5*7*11*13*2*3 = 180180
constexpr auto lengths = make_tuple(Number<2>{}, Number<3>{}, Number<5>{}, Number<7>{},
Number<11>{}, Number<13>{}, Number<2>{}, Number<3>{});
constexpr auto desc = make_naive_tensor_descriptor_packed(lengths);
EXPECT_EQ(desc.GetElementSpaceSize(), 180180);
}
TEST(MakeNaiveTensorDescriptor, ElementSpaceSize8D)
{
// 8D tensor with permuted layout (non-monotonous strides): 2x3x5x7x11x13x17x19
// Memory order: [dim7, dim3, dim1, dim5, dim2, dim0, dim6, dim4]
// This gives strides: [25935, 133, 5187, 19, 881790, 399, 51870, 1]
// Note: strides go up/down/up/down - not monotonously increasing or decreasing
// element_space_size = 2*3*5*7*11*13*17*19 = 9699690
constexpr auto lengths = make_tuple(Number<2>{}, Number<3>{}, Number<5>{}, Number<7>{},
Number<11>{}, Number<13>{}, Number<17>{}, Number<19>{});
constexpr auto strides = make_tuple(Number<25935>{}, Number<133>{}, Number<5187>{}, Number<19>{},
Number<881790>{}, Number<399>{}, Number<51870>{}, Number<1>{});
constexpr auto desc = make_naive_tensor_descriptor(lengths, strides);
EXPECT_EQ(desc.GetElementSpaceSize(), 9699690);
}