提交 a4cc7dbd authored 作者: notoraptor's avatar notoraptor

Flake8 issue corrected.

上级 a4eb981d
......@@ -745,8 +745,8 @@ def blas_header_text():
common_code = code.read()
with open(gemm_template_filepath) as code:
template_code = code.read()
sgemm_code = template_code % {"float_type":"float", "float_size":4, "npy_float":"NPY_FLOAT32", "name":"sgemm_"}
dgemm_code = template_code % {"float_type":"double", "float_size":8, "npy_float":"NPY_FLOAT64", "name":"dgemm_"}
sgemm_code = template_code % {"float_type": "float", "float_size": 4, "npy_float": "NPY_FLOAT32", "name": "sgemm_"}
dgemm_code = template_code % {"float_type": "double", "float_size": 8, "npy_float": "NPY_FLOAT64", "name": "dgemm_"}
if not common_code or not sgemm_code:
raise IOError("Unable to load NumPy implementation of gemm code from C source files.")
else:
......
......@@ -131,7 +131,7 @@ class TestCorr2D(utt.InferShapeTester):
icol:icol + dil_fil_shape2d[1]:filter_dilation[1]] * filter2d[::-1, ::-1]
).sum()
#utt.assert_allclose(theano_output, ref_output)
# utt.assert_allclose(theano_output, ref_output)
utt.assert_allclose(ref_output, theano_output)
# TEST GRADIENT
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论