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 wweic committed May 13, 2019
1 parent 6c362dc commit f9518dc
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 f9518dc

Please sign in to comment.