Function TEST_CASE

test/op/builder/gemm_test.cpp:28–37

#include <migraphx/instruction.hpp>

TEST_CASE(gemm_invalid_input_dim_op_builder_test)
{
    migraphx::module mm;
    mm.add_parameter("a", {migraphx::shape::float_type, {3}});
    mm.add_parameter("b", {migraphx::shape::float_type, {3, 3, 3}});

    EXPECT(test::throws<migraphx::exception>(
        [&] { make_op_module("gemm", {}, mm.get_parameters()); },
        "gemm op_builder: A and B should be rank 2, A is rank 1, B is rank 3"));
}

TEST_CASE(gemm_normal_path_op_builder_test)
{