Skip to content

Commit

Permalink
Change assert condition to fatal.
Browse files Browse the repository at this point in the history
Signed-off-by: Nilaykumar Patel <[email protected]>
  • Loading branch information
nkpatel-tt committed Dec 11, 2024
1 parent 28c64c0 commit 792ae01
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion ttnn/cpp/ttnn/tensor/tensor_utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,9 @@ Tensor create_tensor_from_owned_buffer(
std::move(OwnedStorage{std::move(output_uint32_buffer)}), output_shape, output_dtype, Layout::TILE);
}
} else {
TT_ASSERT((output_dtype != DataType::BFLOAT8_B) || (output_dtype != DataType::BFLOAT4_B));
TT_FATAL(
(output_dtype != DataType::BFLOAT8_B) || (output_dtype != DataType::BFLOAT4_B),
"Unsupported output datatype");
}
auto rm_tensor = Tensor(std::move(OwnedStorage{std::move(buf)}), output_shape, output_dtype, Layout::ROW_MAJOR);
return rm_tensor.to(Layout::TILE);
Expand Down

0 comments on commit 792ae01

Please sign in to comment.