Skip to content

Commit

Permalink
#0: adding new reshape sweeps and correcting existing ones (#15866)
Browse files Browse the repository at this point in the history
### Problem description
Adding the new torch sweeps for reshape

### What's changed
Added torch sweeps to reshape
Added a parser for their new format
Updated reshape.reshape test to test for bfloat16 and float32 rather
than bfloat8 and bfloat16

### Checklist
- [x] Post commit CI passes
https://github.com/tenstorrent/tt-metal/actions/runs/12269839018
- [ ] Blackhole Post commit (if applicable)
- [ ] Model regression CI testing passes (if applicable)
- [ ] Device performance regression CI testing passes (if applicable)
- [ ] **(For models and ops writers)** Full [new
models](https://github.com/tenstorrent/tt-metal/actions/workflows/full-new-models-suite.yaml)
tests passes
- [ ] New/Existing tests provide coverage for changes
  • Loading branch information
jvegaTT authored Dec 11, 2024
1 parent 8e49222 commit 13f512a
Show file tree
Hide file tree
Showing 5 changed files with 5,271 additions and 3 deletions.
1 change: 1 addition & 0 deletions .github/workflows/ttnn-run-sweeps.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -356,6 +356,7 @@ on:
- data_movement.squeeze.squeeze_pytorch2
- data_movement.embedding.embedding_pytorch2
- data_movement.view.view_pytorch2
- data_movement.view.view_tt_torch
schedule:
- cron: "0 21 * * *" # This cron schedule runs the workflow at 9:00pm UTC nightly

Expand Down
4 changes: 2 additions & 2 deletions tests/sweep_framework/sweeps/data_movement/reshape/reshape.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,8 +67,8 @@ def gen_reshape_shape(input_shape, step=1):
"input_shape": gen_shapes([1, 1, 1, 1], [6, 6, 256, 256], [1, 1, 1, 1], 16)
+ gen_shapes([1, 1, 1], [6, 256, 256], [1, 1, 1], 16)
+ gen_shapes([1, 1], [256, 256], [1, 1], 16),
"input_a_dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
"input_a_layout": [ttnn.TILE_LAYOUT], # ttnn.ROW_MAJOR_LAYOUT
"input_a_dtype": [ttnn.bfloat16, ttnn.float32],
"input_a_layout": [ttnn.TILE_LAYOUT, ttnn.ROW_MAJOR_LAYOUT], # ttnn.ROW_MAJOR_LAYOUT
"input_a_memory_config": [ttnn.DRAM_MEMORY_CONFIG, ttnn.L1_MEMORY_CONFIG],
},
}
Expand Down
Loading

0 comments on commit 13f512a

Please sign in to comment.