Skip to content

Commit 150554d

Browse files
bani-intelaipgsayantan-nervana
authored andcommitted
Bani/merge NGTF-2400 BatchMatMulV2-fix into r0.19 (#457)
1 parent aee7098 commit 150554d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ngraph_bridge/ngraph_builder.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -977,10 +977,10 @@ static Status TranslateBatchMatMulV2Op(
977977
"for each dimension. Failed to match dimension ",
978978
i, " where lhs was ", ng_lhs_shape[i], " and rhs was ",
979979
ng_rhs_shape[i]);
980-
} else if ((ng_lhs_shape[i] == 1) || (ng_rhs_shape[i] == 1)) {
980+
} else if (ng_lhs_shape[i] != ng_lhs_shape[i]) {
981981
return errors::Unimplemented(
982982
"ng_lhs_shape and ng_rhs_shape must be the same for each dimension, "
983-
"for current implementation of BatchMatMulV2."
983+
"for current implementation of BatchMatMulV2. "
984984
"Failed to match dimension ",
985985
i, " where lhs was ", ng_lhs_shape[i], " and rhs was ",
986986
ng_rhs_shape[i]);

0 commit comments

Comments
 (0)