Skip to content

Commit

Permalink
[Relay] Fix BatchMatMulRel typerelation (apache#3032)
Browse files Browse the repository at this point in the history
return false mean retry in the future, and in the case of error, it should be report ASAP, not retry.
  • Loading branch information
MarisaKirisame authored and Wei Chen committed May 13, 2019
1 parent 1d71097 commit 295b33d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/relay/op/nn/nn.cc
Original file line number Diff line number Diff line change
Expand Up @@ -658,7 +658,7 @@ bool BatchMatmulRel(const Array<Type>& types,
const auto* x = types[0].as<TensorTypeNode>();
const auto* y = types[1].as<TensorTypeNode>();
if (x == nullptr || y == nullptr) return false;
if (x->shape.size() != 3 || y->shape.size() != 3) return false;
CHECK(x->shape.size() == 3 && y->shape.size() == 3);
CHECK(reporter->AssertEQ(x->shape[0], y->shape[0]))
<< "BatchDot: batch dimension doesn't match, "
<< " x shape=" << x->shape
Expand Down

0 comments on commit 295b33d

Please sign in to comment.