Skip to content

Commit

Permalink
#0: Improve matmul error message when inner dimensions do not match
Browse files Browse the repository at this point in the history
  • Loading branch information
esmalTT committed Oct 16, 2024
1 parent 7b3d1b3 commit dfc7299
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ttnn/cpp/ttnn/operations/matmul/matmul.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ ttnn::Tensor bound_matmul(
const auto height_b = input_tensor_b_shape[-2];

if (width_a != height_b) {
TT_THROW("ttnn.matmul: The width of the first tensor must be equal to the height of the second tensor");
TT_THROW("ttnn.matmul: The width of the first tensor must be equal to the height of the second tensor ({} != {}). The shape of first tensor was {} and the shape of second tensor was {})", width_a, height_b, input_tensor_a_shape, input_tensor_b_shape);
}

const bool has_program_config = parameters.program_config.has_value();
Expand Down

0 comments on commit dfc7299

Please sign in to comment.