Skip to content

Commit

Permalink
#0: Addded comments to the file
Browse files Browse the repository at this point in the history
* Addresses issue #18064 (more specifically #16439)
  • Loading branch information
ncvetkovicTT committed Feb 20, 2025
1 parent 2950b6e commit bbcf441
Showing 1 changed file with 14 additions and 8 deletions.
22 changes: 14 additions & 8 deletions tests/tt_metal/tt_metal/integration/matmul/test_matmul_hang.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -213,12 +213,10 @@ using namespace tt::test_utils;
using namespace unit_tests_common::matmul::test_matmul_blackhole_hang;

TEST_F(DispatchFixture, TestMatmulBlackholeHangCombined) {
uint32_t M1 = 2;
uint32_t K1 = 4;
uint32_t N1 = 4;
uint32_t M2 = 2;
uint32_t K2 = 4;
uint32_t N2 = 8;
// First iteration setup
uint32_t M1 = 2; /* M = 512 from test_benchmark */
uint32_t K1 = 4; /* K = 1024 from test_benchmark */
uint32_t N1 = 4; /* N = 1024 from test_benchmark */
MatmulHangConfig matmul_config1 = {
.M = M1,
.K = K1,
Expand Down Expand Up @@ -248,6 +246,13 @@ TEST_F(DispatchFixture, TestMatmulBlackholeHangCombined) {
0, /* untilize_out */
},
.math_fidelity = MathFidelity(4)};
MatmulHangStimuli stimuli1;
create_test_stimuli(stimuli1, M1, K1, N1);

// Second iteration setup
uint32_t M2 = 2; /* M = 512 from test_benchmark */
uint32_t K2 = 4; /* K = 1024 from test_benchmark */
uint32_t N2 = 8; /* N = 2048 from test_benchmark */
MatmulHangConfig matmul_config2 = {
.M = M2,
.K = K2,
Expand Down Expand Up @@ -277,10 +282,11 @@ TEST_F(DispatchFixture, TestMatmulBlackholeHangCombined) {
0, /* untilize_out */
},
.math_fidelity = MathFidelity(4)};
MatmulHangStimuli stimuli1;
create_test_stimuli(stimuli1, M1, K1, N1);
MatmulHangStimuli stimuli2;
create_test_stimuli(stimuli2, M2, K2, N2);

// First iteration execution
matmul_tile(this, devices_.at(0), matmul_config1, stimuli1.a, stimuli1.w, stimuli1.t);
// Second iteration execution
matmul_tile(this, devices_.at(0), matmul_config2, stimuli2.a, stimuli2.w, stimuli2.t);
}

0 comments on commit bbcf441

Please sign in to comment.