AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use the precalculated num_warps variable

Browse Source
This commit is contained in:
Yiltan Hassan Temucin
2025-02-06 13:12:09 -06:00
parent 8d74c7b73e
commit b83ff2fa84
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/functional_tests/driver.sh
+2 -2
View File
@@ -270,13 +270,13 @@ TestSigOps() {
ExecTest "putsignal" 2 2 32 1048576
ExecTest "wgputsignal" 2 2 32 1048576
ExecTest "waveputsignal" 2 1 32 1048576
ExecTest "waveputsignal" 2 1 64 1048576
ExecTest "waveputsignal" 2 2 64 1048576
ExecTest "putsignalnbi" 2 1 1 1048576
ExecTest "putsignalnbi" 2 2 32 1048576
ExecTest "wgputsignalnbi" 2 2 32 1048576
ExecTest "waveputsignalnbi" 2 1 32 1048576
ExecTest "waveputsignalnbi" 2 1 64 1048576
ExecTest "waveputsignalnbi" 2 2 64 1048576
ExecTest "signalfetch" 2 1 1
ExecTest "wgsignalfetch" 2 2 32
tests/functional_tests/signaling_operations_tester.cpp
+1 -3
View File
@@ -201,7 +201,6 @@ void SignalingOperationsTester::verifyResults(uint64_t size) {
} else if (ROCSHMEM_SIGNAL_ADD == sig_op) {
uint64_t value = *sig_addr;
uint64_t expected_value = (args.myid + 123); // Initial Value
uint64_t num_waves = 1;
switch (_type) {
case PutSignalTestType:
@@ -214,8 +213,7 @@ void SignalingOperationsTester::verifyResults(uint64_t size) {
break;
case WAVEPutSignalTestType:
case WAVEPutSignalNBITestType:
num_waves = max(1, (args.num_wgs / __AMDGCN_WAVEFRONT_SIZE__));
expected_value += ((args.skip + args.loop) * num_waves);
expected_value += ((args.skip + args.loop) * args.num_wgs * num_warps);
break;
default:
fprintf(stderr, "Invalid Test\n");