提交 2880cb32 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Do not use blas routines in C if no blas present

上级 48e862bd
...@@ -51,7 +51,7 @@ class Conv3D(theano.Op): ...@@ -51,7 +51,7 @@ class Conv3D(theano.Op):
return "Conv3D" return "Conv3D"
def c_code_cache_version(self): def c_code_cache_version(self):
return (2,) return (3,)
def make_node(self, V, W, b, d): def make_node(self, V, W, b, d):
...@@ -338,7 +338,8 @@ class Conv3D(theano.Op): ...@@ -338,7 +338,8 @@ class Conv3D(theano.Op):
#if the data types are not mixed, we can insert special case optimizations based on BLAS #if the data types are not mixed, we can insert special case optimizations based on BLAS
VV, WV, bv, dv = node.inputs VV, WV, bv, dv = node.inputs
HV = node.outputs[0] HV = node.outputs[0]
if VV.dtype == WV.dtype and HV.dtype == VV.dtype: if (theano.config.blas.ldflags and
VV.dtype == WV.dtype and HV.dtype == VV.dtype):
if VV.dtype == 'float64': if VV.dtype == 'float64':
gemv = 'dgemv_' gemv = 'dgemv_'
elif VV.dtype == 'float32': elif VV.dtype == 'float32':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论