Expand tensor descriptor test coverage

- Add tests for make_naive_tensor_descriptor_packed (1D, 2D, 3D)
- Add tests for make_naive_tensor_descriptor_aligned (2D, 3D)
- Add 1D tensor tests with explicit strides
- Ensure all shapes use distinct, coprime dimensions

test/util/unit_tensor_descriptor_helper.cpp

@@ -53,40 +53,110 @@ TEST(MakeNaiveTensorDescriptor, WithPaddingArbitrary)
 
 TEST(MakeNaiveTensorDescriptor, WithPaddingStrideSlice)
 {
-    // 2x3x4 tensor with strides [1, 5, 35] - like a slice from a 5x7xN column-major tensor
+    // 2x3x5 tensor with strides [1, 7, 35] - like a slice from a 7x7xN column-major tensor
     // This tests padding where there's space for extra elements
-    // element_space_size = 1 + (2-1)*1 + (3-1)*5 + (4-1)*35 = 1 + 1 + 10 + 105 = 117
-    constexpr auto lengths = make_tuple(Number<2>{}, Number<3>{}, Number<4>{});
-    constexpr auto strides = make_tuple(Number<1>{}, Number<5>{}, Number<35>{});
+    // element_space_size = 1 + (2-1)*1 + (3-1)*7 + (5-1)*35 = 1 + 1 + 14 + 140 = 156
+    constexpr auto lengths = make_tuple(Number<2>{}, Number<3>{}, Number<5>{});
+    constexpr auto strides = make_tuple(Number<1>{}, Number<7>{}, Number<35>{});
     constexpr auto desc    = make_naive_tensor_descriptor(lengths, strides);
-    EXPECT_EQ(desc.GetElementSpaceSize(), 117);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 156);
 }
 
 TEST(MakeNaiveTensorDescriptor, ColumnMajor)
 {
-    // 5x4 tensor with column-major strides [1, 5]
-    // element_space_size = 1 + (5-1)*1 + (4-1)*5 = 1 + 4 + 15 = 20
-    constexpr auto lengths = make_tuple(Number<5>{}, Number<4>{});
-    constexpr auto strides = make_tuple(Number<1>{}, Number<5>{});
+    // 7x3 tensor with column-major strides [1, 7]
+    // element_space_size = 1 + (7-1)*1 + (3-1)*7 = 1 + 6 + 14 = 21
+    constexpr auto lengths = make_tuple(Number<7>{}, Number<3>{});
+    constexpr auto strides = make_tuple(Number<1>{}, Number<7>{});
     constexpr auto desc    = make_naive_tensor_descriptor(lengths, strides);
-    EXPECT_EQ(desc.GetElementSpaceSize(), 20);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 21);
 }
 
 // Test with runtime values (index_t instead of Number<>)
 TEST(MakeNaiveTensorDescriptorRuntime, Simple2D)
 {
-    // 5x4 tensor with row-major strides
-    const auto lengths = make_tuple(index_t{5}, index_t{4});
+    // 9x4 tensor with row-major strides
+    // element_space_size = 1 + (9-1)*4 + (4-1)*1 = 1 + 32 + 3 = 36
+    const auto lengths = make_tuple(index_t{9}, index_t{4});
     const auto strides = make_tuple(index_t{4}, index_t{1});
     const auto desc    = make_naive_tensor_descriptor(lengths, strides);
-    EXPECT_EQ(desc.GetElementSpaceSize(), 20);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 36);
 }
 
 TEST(MakeNaiveTensorDescriptorRuntime, WithPadding)
 {
-    // 11x7x3 tensor with arbitrary strides (using prime numbers)
-    const auto lengths = make_tuple(index_t{11}, index_t{7}, index_t{3});
+    // 13x5x2 tensor with arbitrary strides (using prime numbers)
+    // element_space_size = 1 + (13-1)*2 + (5-1)*97 + (2-1)*23 = 1 + 24 + 388 + 23 = 436
+    const auto lengths = make_tuple(index_t{13}, index_t{5}, index_t{2});
     const auto strides = make_tuple(index_t{2}, index_t{97}, index_t{23});
     const auto desc    = make_naive_tensor_descriptor(lengths, strides);
-    EXPECT_EQ(desc.GetElementSpaceSize(), 649);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 436);
+}
+
+// Test 1D tensors with explicit strides
+TEST(MakeNaiveTensorDescriptor, ElementSpaceSize1D)
+{
+    // 13-element 1D tensor with stride 1
+    constexpr auto lengths = make_tuple(Number<13>{});
+    constexpr auto strides = make_tuple(Number<1>{});
+    constexpr auto desc    = make_naive_tensor_descriptor(lengths, strides);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 13);
+}
+
+TEST(MakeNaiveTensorDescriptor, ElementSpaceSize1DStrided)
+{
+    // 7-element 1D tensor with stride 3 (every 3rd element)
+    // element_space_size = 1 + (7-1)*3 = 19
+    constexpr auto lengths = make_tuple(Number<7>{});
+    constexpr auto strides = make_tuple(Number<3>{});
+    constexpr auto desc    = make_naive_tensor_descriptor(lengths, strides);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 19);
+}
+
+// Test make_naive_tensor_descriptor_packed (contiguous layout)
+// element_space_size = product of all lengths
+TEST(MakeNaiveTensorDescriptorPacked, Simple1D)
+{
+    // 17-element packed tensor - element_space_size = 17
+    constexpr auto lengths = make_tuple(Number<17>{});
+    constexpr auto desc    = make_naive_tensor_descriptor_packed(lengths);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 17);
+}
+
+TEST(MakeNaiveTensorDescriptorPacked, Simple2D)
+{
+    // 6x5 packed tensor - element_space_size = 6*5 = 30
+    constexpr auto lengths = make_tuple(Number<6>{}, Number<5>{});
+    constexpr auto desc    = make_naive_tensor_descriptor_packed(lengths);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 30);
+}
+
+TEST(MakeNaiveTensorDescriptorPacked, Simple3D)
+{
+    // 4x5x9 packed tensor - element_space_size = 4*5*9 = 180
+    constexpr auto lengths = make_tuple(Number<4>{}, Number<5>{}, Number<9>{});
+    constexpr auto desc    = make_naive_tensor_descriptor_packed(lengths);
+    EXPECT_EQ(desc.GetElementSpaceSize(), 180);
+}
+
+// Test make_naive_tensor_descriptor_aligned (stride alignment for memory access)
+// Aligns the second-to-last stride to be a multiple of 'align'
+TEST(MakeNaiveTensorDescriptorAligned, Align4)
+{
+    // 5x3 tensor aligned to 4 elements
+    // strides[1] = 1, strides[0] = integer_least_multiple(3, 4) = 4
+    // element_space_size = 1 + (5-1)*4 + (3-1)*1 = 1 + 16 + 2 = 19
+    constexpr auto lengths = make_tuple(Number<5>{}, Number<3>{});
+    constexpr auto desc    = make_naive_tensor_descriptor_aligned(lengths, Number<4>{});
+    EXPECT_EQ(desc.GetElementSpaceSize(), 19);
+}
+
+TEST(MakeNaiveTensorDescriptorAligned, Align8)
+{
+    // 3x5x7 tensor aligned to 8 elements
+    // strides[2] = 1, strides[1] = integer_least_multiple(7, 8) = 8, strides[0] = 5*8 = 40
+    // element_space_size = 1 + (3-1)*40 + (5-1)*8 + (7-1)*1 = 1 + 80 + 32 + 6 = 119
+    constexpr auto lengths = make_tuple(Number<3>{}, Number<5>{}, Number<7>{});
+    constexpr auto desc    = make_naive_tensor_descriptor_aligned(lengths, Number<8>{});
+    EXPECT_EQ(desc.GetElementSpaceSize(), 119);
 }