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

Add tilled version of puts and gets at the workgroup level to the functional test suite

Browse Source 
[ROCm/rocshmem commit: d226922733]
This commit is contained in:
avinashkethineedi
2024-09-04 12:55:10 -07:00
parent 31c43bd58d
commit 3d26792831
6 changed files with 89 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/tests/functional_tests/tester.cpp
+16 -4
View File
@@ -469,19 +469,31 @@ std::vector<Tester*> Tester::create(TesterArguments args) {
testers.push_back(new ShmemPtrTester(args));
return testers;
case WGGetTestType:
if (rank == 0) std::cout << "Blocking WG level Gets***" << std::endl;
if (rank == 0) {
if (args.tiled) std::cout << "Tiled Blocking WG level Gets***" << std::endl;
else std::cout << "Blocking WG level Gets***" << std::endl;
}
testers.push_back(new ExtendedPrimitiveTester(args));
return testers;
case WGGetNBITestType:
if (rank == 0) std::cout << "Non-Blocking WG level Gets***" << std::endl;
if (rank == 0) {
if (args.tiled) std::cout << "Tiled Non-Blocking WG level Gets***" << std::endl;
else std::cout << "Non-Blocking WG level Gets***" << std::endl;
}
testers.push_back(new ExtendedPrimitiveTester(args));
return testers;
case WGPutTestType:
if (rank == 0) std::cout << "Blocking WG level Puts***" << std::endl;
if (rank == 0) {
if (args.tiled) std::cout << "Tiled Blocking WG level Puts***" << std::endl;
else std::cout << "Blocking WG level Puts***" << std::endl;
}
testers.push_back(new ExtendedPrimitiveTester(args));
return testers;
case WGPutNBITestType:
if (rank == 0) std::cout << "Non-Blocking WG level Puts***" << std::endl;
if (rank == 0) {
if(args.tiled) std::cout << "Tiled Non-Blocking WG level Puts***" << std::endl;
else std::cout << "Non-Blocking WG level Puts***" << std::endl;
}
testers.push_back(new ExtendedPrimitiveTester(args));
return testers;
case PutNBIMRTestType: