Commit b85b8559 authored by Paul's avatar Paul
Browse files

Disable nonpacked test for now

parent 647d7908
...@@ -14,10 +14,9 @@ static auto make_mat(tensor_view<T> x) ...@@ -14,10 +14,9 @@ static auto make_mat(tensor_view<T> x)
{ {
const auto& s = x.get_shape(); const auto& s = x.get_shape();
assert(s.lens().size() == 2); assert(s.lens().size() == 2);
assert(s.packed());
if(s.transposed()) if(s.transposed())
return matrix<T>{x.data(), s.lens()[1], s.lens()[0]}; return matrix<T>{x.data(), s.lens()[1], s.lens()[0], s.strides()[1]};
return matrix<T>{x.data(), s.lens()[0], s.lens()[1]}; return matrix<T>{x.data(), s.lens()[0], s.lens()[1], s.strides()[0]};
} }
template <class T, class F> template <class T, class F>
......
...@@ -141,6 +141,18 @@ struct test_gemm ...@@ -141,6 +141,18 @@ struct test_gemm
} }
}; };
struct test_gemm_ld
{
migraph::program create_program() const
{
migraph::program p;
auto a = p.add_parameter("a", migraph::shape{migraph::shape::float_type, {4, 5}, {10, 1}});
auto b = p.add_parameter("b", migraph::shape{migraph::shape::float_type, {5, 3}, {20, 1}});
p.add_instruction(migraph::gemm{}, a, b);
return p;
}
};
struct test_gemm_transposeb struct test_gemm_transposeb
{ {
migraph::program create_program() const migraph::program create_program() const
...@@ -214,6 +226,7 @@ int main() ...@@ -214,6 +226,7 @@ int main()
verify_program<test_conv_relu>(); verify_program<test_conv_relu>();
verify_program<test_conv_pooling>(); verify_program<test_conv_pooling>();
verify_program<test_gemm>(); verify_program<test_gemm>();
// verify_program<test_gemm_ld>();
verify_program<test_gemm_transposeb>(); verify_program<test_gemm_transposeb>();
verify_program<test_gemm_transposea>(); verify_program<test_gemm_transposea>();
verify_program<test_gemm_transposeab>(); verify_program<test_gemm_transposeab>();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment