提交 435caa76 authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't use CpuCorrMM when there is no CPU BLAS

上级 8a736df5
......@@ -62,7 +62,7 @@ compile.optdb.register('local_inplace_sparse_block_outer',
# Conv opts
@local_optimizer([AbstractConv2d])
def local_abstractconv_gemm(node):
if theano.config.cxx == "":
if theano.config.cxx == "" or theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d):
return None
......@@ -83,7 +83,7 @@ def local_abstractconv_gemm(node):
@local_optimizer([AbstractConv2d_gradWeights])
def local_abstractconv_gradweight_gemm(node):
if theano.config.cxx == "":
if theano.config.cxx == "" or theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d_gradWeights):
return None
......@@ -107,7 +107,7 @@ def local_abstractconv_gradweight_gemm(node):
@local_optimizer([AbstractConv2d_gradInputs])
def local_abstractconv_gradinputs_gemm(node):
if theano.config.cxx == "":
if theano.config.cxx == "" or theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d_gradInputs):
return None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论