提交 d9a522a5 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add tests for CGemv.

They should detect the bug described in gh-1240
上级 eacc8509
...@@ -208,8 +208,12 @@ class TestCGemv(TestCase, TestOptimizationMixin): ...@@ -208,8 +208,12 @@ class TestCGemv(TestCase, TestOptimizationMixin):
def test_gemv1(self): def test_gemv1(self):
self.t_gemv1((3, 2)) self.t_gemv1((3, 2))
self.t_gemv1((1, 2))
self.t_gemv1((0, 2)) self.t_gemv1((0, 2))
self.t_gemv1((3, 1))
self.t_gemv1((3, 0)) self.t_gemv1((3, 0))
self.t_gemv1((1, 0))
self.t_gemv1((0, 1))
self.t_gemv1((0, 0)) self.t_gemv1((0, 0))
def test_gemv_dimensions(self, dtype='float32'): def test_gemv_dimensions(self, dtype='float32'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论