Skip to content

Commit

Permalink
tilize_op float32 access (#15115)
Browse files Browse the repository at this point in the history
### Ticket
[Link to Github
Issue](#14885 (comment))

### Problem description
I missed one entry point for tilize when adding float32 support earlier.

### What's changed
- Just added the one entry point in tilize_op.cpp, wanted to push this
out real quick since it's a quick fix and there seems to be a need for
it!

### Checklist
- [ ] Post commit CI passes:
https://github.com/tenstorrent/tt-metal/actions/runs/11862518244
  • Loading branch information
yugi957 authored Nov 18, 2024
1 parent 60402d0 commit 55912dc
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ void Tilize::validate(const std::vector<Tensor>& input_tensors) const {
auto width = input_tensor_a.get_legacy_shape()[-1];
uint32_t stick_s = width;
uint32_t num_sticks = input_tensor_a.volume() / width;
TT_FATAL(input_tensor_a.get_dtype() == DataType::BFLOAT16, "Error");
TT_FATAL(input_tensor_a.get_dtype() == DataType::BFLOAT16 or input_tensor_a.get_dtype() == DataType::FLOAT32, "data type must be bfloat16 or float32");

uint32_t stick_size = stick_s * input_tensor_a.element_size(); // Assuming bfloat16 dataformat

Expand Down

0 comments on commit 55912dc

Please sign in to comment.