-
Notifications
You must be signed in to change notification settings - Fork 98
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Removing skip_for_blackhole and fix some minor issues for blackhole conv2d #17222
base: main
Are you sure you want to change the base?
Conversation
@@ -578,7 +579,7 @@ def test_conv_ws( | |||
auto_shard, | |||
tilized_input, | |||
): | |||
if device.core_grid.y != 8: | |||
if device.core_grid.y != 8 and is_wormhole_b0(): | |||
pytest.skip("Needs 8x8 Grid") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this comment doesn't make sense anymore
@@ -983,6 +983,8 @@ conv_op_l1_usage conv2d::calculate_L1_usage( | |||
(per_core_out_matrix_height_ntiles + act_block_h_ntiles - 1) / act_block_h_ntiles; | |||
uint32_t out_block_h_ntiles_padded = num_blocks_act_h_per_core * act_block_h_ntiles; | |||
|
|||
uint32_t alignment_bytes = arch == tt::ARCH::BLACKHOLE ? 64 : 32; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Get l1 aligment from HAL
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You will need to disable shallow convs for BH in auto shard codepath.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
18 of them fail because of an assert conv2d_op_width_sharded_program_factory.cpp specify which assert is problematic and post new issue here.
c5b8abc
to
1733874
Compare
Issues
#17221
#17216
Problem description
Enable conv2d blackhole tests - to begin with, remove the skips in
tests/ttnn/unit_tests/operations/test_new_conv2d.py
and provide a report about state of tests. Fix minor bugs described in issues.What's changed
@skip_for_blackhole()
for tests that already run on wormhole_b0calculate_L1_usage
functionState of blackhole tests after this changes
repro:
pytest tests/ttnn/unit_tests/operations/test_new_conv2d.py
PASSED: 2244
FAILED: 65
97% pass rate
Of the tests that fail:
47 of them fail because of PCC error
18 of them fail because of an assert
conv2d_op_width_sharded_program_factory.cpp
(these problems are going to be tracked in proper issues)
Checklist