提交 ad231afe authored 作者: Reyhane Askari's avatar Reyhane Askari

minor fix

上级 83a8b1f4
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import unittest
import theano import theano
from theano import tensor from theano import tensor
from theano.tests.unittest_tools import assertFailure_fast
from theano.gof.opt import check_stack_trace from theano.gof.opt import check_stack_trace
from theano.tensor.nnet.blocksparse import ( from theano.tensor.nnet.blocksparse import (
sparse_block_dot, sparse_block_gemv_inplace, sparse_block_outer_inplace, sparse_block_dot, sparse_block_gemv_inplace, sparse_block_outer_inplace,
...@@ -26,8 +26,8 @@ def test_blocksparse_inplace_gemv_opt(): ...@@ -26,8 +26,8 @@ def test_blocksparse_inplace_gemv_opt():
assert f.maker.fgraph.toposort()[-1].op.inplace assert f.maker.fgraph.toposort()[-1].op.inplace
assert check_stack_trace(f, ops_to_check=[sparse_block_gemv_inplace]) assert check_stack_trace(f, ops_to_check=[sparse_block_gemv_inplace])
if theano.config.cycle_detection == 'fast' and theano.config.mode != 'FAST_COMPILE': if theano.config.mode != 'FAST_COMPILE':
test_blocksparse_inplace_gemv_opt = unittest.expectedFailure(test_blocksparse_inplace_gemv_opt) test_blocksparse_inplace_gemv_opt = assertFailure_fast(test_blocksparse_inplace_gemv_opt)
def test_blocksparse_inplace_outer_opt(): def test_blocksparse_inplace_outer_opt():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论