提交 b1877db5 authored 作者: Frederic's avatar Frederic

fix blas_c tests due to wrong mode(default exclude c blas optimization)

上级 b5737084
...@@ -17,6 +17,10 @@ from theano.tests.unittest_tools import TestOptimizationMixin ...@@ -17,6 +17,10 @@ from theano.tests.unittest_tools import TestOptimizationMixin
from test_blas import TestCase from test_blas import TestCase
from test_blas import BaseGemv from test_blas import BaseGemv
mode_blas_opt = theano.compile.get_default_mode().including(
'BlasOpt', 'specialize', 'InplaceBlasOpt')
class TestCGer(TestCase, TestOptimizationMixin): class TestCGer(TestCase, TestOptimizationMixin):
def setUp(self, dtype='float64'): def setUp(self, dtype='float64'):
...@@ -202,12 +206,14 @@ class TestCGemv(TestCase, TestOptimizationMixin): ...@@ -202,12 +206,14 @@ class TestCGemv(TestCase, TestOptimizationMixin):
class TestCGemvFloat32(TestCase, BaseGemv, TestOptimizationMixin): class TestCGemvFloat32(TestCase, BaseGemv, TestOptimizationMixin):
mode = mode_blas_opt
dtype = 'float32' dtype = 'float32'
gemv = CGemv(inplace=False) gemv = CGemv(inplace=False)
gemv_inplace = CGemv(inplace=True) gemv_inplace = CGemv(inplace=True)
class TestCGemvFloat64(TestCase, BaseGemv, TestOptimizationMixin): class TestCGemvFloat64(TestCase, BaseGemv, TestOptimizationMixin):
mode = mode_blas_opt
dtype = 'float64' dtype = 'float64'
gemv = CGemv(inplace=False) gemv = CGemv(inplace=False)
gemv_inplace = CGemv(inplace=True) gemv_inplace = CGemv(inplace=True)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论