Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix linux-IC error
Browse files Browse the repository at this point in the history
jellyman2187 committed Oct 30, 2021
1 parent 47a62b9 commit 223f7fc
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions framework/api/nn/test_beamsearchdecoder.py
Original file line number Diff line number Diff line change
@@ -388,8 +388,8 @@ def test_beamsearchdecoder6():
try:
dynamic_decode(decoder=decoder, inits=decoder_cell.get_initial_states(encoder_output), max_step_num=5)
except Exception as e:
# print(e)
if "[operator < matmul > error]" in e.args[0]:
print(e)
if ("[operator < matmul > error]" in e.args[0]) or ("[operator < matmul_v2 > error]" in e.args[0]):
pass
else:
raise Exception

0 comments on commit 223f7fc

Please sign in to comment.