提交 6b3afd89 authored 作者: notoraptor's avatar notoraptor

flake8 errors fixed

上级 39a2b2d2
......@@ -981,9 +981,8 @@ def blas_header_text():
}
""")
header += gemm_code
return header % {'const':const}
return header % {'const': const}
def mkl_threads_text():
......
......@@ -72,7 +72,7 @@ compile.optdb.register('local_inplace_sparse_block_outer',
# Conv opts
@local_optimizer([AbstractConv2d])
def local_abstractconv_gemm(node):
if theano.config.cxx == "": # or not theano.config.blas.ldflags:
if theano.config.cxx == "": # or not theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d):
return None
......@@ -116,7 +116,7 @@ def local_abstractconv3d_gemm(node):
@local_optimizer([AbstractConv2d_gradWeights])
def local_abstractconv_gradweight_gemm(node):
if theano.config.cxx == "" : # or not theano.config.blas.ldflags:
if theano.config.cxx == "": # or not theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d_gradWeights):
return None
......@@ -166,7 +166,7 @@ def local_abstractconv3d_gradweight_gemm(node):
@local_optimizer([AbstractConv2d_gradInputs])
def local_abstractconv_gradinputs_gemm(node):
if theano.config.cxx == "" : # or not theano.config.blas.ldflags:
if theano.config.cxx == "": # or not theano.config.blas.ldflags:
return
if not isinstance(node.op, AbstractConv2d_gradInputs):
return None
......
......@@ -419,7 +419,7 @@ class TestCorrConv2d(BaseTestConv2d):
def tcase(self, i, f, s, b, flip, provide_shape, fd=(1, 1)):
o = self.get_output_shape(i, f, s, b, fd)
#if (not theano.config.blas.ldflags or
# if (not theano.config.blas.ldflags or
if (not theano.config.cxx or
theano.config.mode == "FAST_COMPILE"):
raise SkipTest("Need blas to test conv2d")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论