Function TEST_F

tests/test_tensor_matrix.cpp:71–95

// ============= Matrix Multiplication Tests =============

TEST_F(TensorMatrixTest, MatMul2D) {
    // Test basic 2D matrix multiplication: (2x3) @ (3x2) = (2x2)
    std::vector<float> data_a = {1, 2, 3,
                                 4, 5, 6}; // 2x3
    std::vector<float> data_b = {7, 8,
                                 9, 10,
                                 11, 12}; // 3x2

    auto custom_a = Tensor::from_vector(data_a, {2, 3}, Device::CUDA);
    auto custom_b = Tensor::from_vector(data_b, {3, 2}, Device::CUDA);

    auto torch_a = torch::tensor(data_a, torch::TensorOptions().device(torch::kCUDA))
                       .reshape({2, 3});
    auto torch_b = torch::tensor(data_b, torch::TensorOptions().device(torch::kCUDA))
                       .reshape({3, 2});

    auto custom_result = custom_a.matmul(custom_b);
    auto torch_result = torch::matmul(torch_a, torch_b);

    compare_tensors(custom_result, torch_result, 1e-4f, 1e-5f, "MatMul2D");

    // Test mm() alias - should give same result
    auto custom_mm = custom_a.mm(custom_b);
    compare_tensors(custom_mm, torch_result, 1e-4f, 1e-5f, "MM_Alias");
}

TEST_F(TensorMatrixTest, MatMulVectorMatrix) {
    // Test vector-matrix multiplication: (3,) @ (3x2) = (2,)