Function TEST_F

tests/test_slice_mm_bug.cpp:100–142

Test 1: Basic slice then contiguous

// Test 1: Basic slice then contiguous
TEST_F(SliceMMBugTest, SliceContiguous_3x3_From_4x4) {
    // Create a 4x4 matrix with known values
    // Row-major: [[1,2,3,4], [5,6,7,8], [9,10,11,12], [13,14,15,16]]
    std::vector<float> data(16);
    for (int i = 0; i < 16; i++) {
        data[i] = static_cast<float>(i + 1);
    }

    auto custom = Tensor::from_vector(data, {4, 4}, Device::CPU);
    auto torch_t = torch::tensor(data).reshape({4, 4});

    // Extract 3x3 submatrix: [:3, :3]
    auto custom_slice = custom.slice(0, 0, 3).slice(1, 0, 3);
    auto torch_slice = torch_t.slice(0, 0, 3).slice(1, 0, 3);

    LOG_INFO("Custom slice shape: {}x{}", custom_slice.shape()[0], custom_slice.shape()[1]);
    LOG_INFO("Custom slice strides: [{}, {}]", custom_slice.stride(0), custom_slice.stride(1));
    LOG_INFO("Custom slice is_contiguous: {}", custom_slice.is_contiguous());

    // Make contiguous
    auto custom_contiguous = custom_slice.contiguous();
    auto torch_contiguous = torch_slice.contiguous();

    LOG_INFO("Custom contiguous strides: [{}, {}]", custom_contiguous.stride(0), custom_contiguous.stride(1));
    LOG_INFO("Custom contiguous is_contiguous: {}", custom_contiguous.is_contiguous());

    // Verify values
    auto custom_vec = custom_contiguous.to_vector();

    LOG_INFO("Custom values: [{}, {}, {}, {}, {}, {}, {}, {}, {}]",
             custom_vec[0], custom_vec[1], custom_vec[2],
             custom_vec[3], custom_vec[4], custom_vec[5],
             custom_vec[6], custom_vec[7], custom_vec[8]);

    // Expected: [[1,2,3], [5,6,7], [9,10,11]]
    std::vector<float> expected = {1, 2, 3, 5, 6, 7, 9, 10, 11};
    for (size_t i = 0; i < expected.size(); i++) {
        EXPECT_FLOAT_EQ(custom_vec[i], expected[i])
            << "Mismatch at index " << i << ": got " << custom_vec[i] << ", expected " << expected[i];
    }

    compare_tensors(custom_contiguous, torch_contiguous, 1e-5f, 1e-6f, "Slice3x3");
}

// Test 2: Extract column slice (translation vector)
TEST_F(SliceMMBugTest, SliceColumn_3x1_From_4x4) {