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

add functional tests for puts and gets at wavefront level

Browse Source 
* These functional tests are simple puts and gets, where every wave will get/put the same amount of data
* Enabled workgroup level puts and gets tests
This commit is contained in:
avinashkethineedi
2024-09-03 14:19:01 -07:00
parent 7bbf34d334
commit ff954237dd
5 changed files with 208 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/functional_tests/tester.cpp
+17
View File
@@ -52,6 +52,7 @@
#include "team_ctx_infra_tester.hpp"
#include "team_ctx_primitive_tester.hpp"
#include "team_reduction_tester.hpp"
#include "wave_level_primitives.hpp"
Tester::Tester(TesterArguments args) : args(args) {
_type = (TestType)args.algorithm;
@@ -488,6 +489,22 @@ std::vector<Tester*> Tester::create(TesterArguments args) {
std::cout << "Non-Blocking Put message rate***" << std::endl;
testers.push_back(new PrimitiveMRTester(args));
return testers;
case WAVEGetTestType:
if (rank == 0) std::cout << "WAVE Blocking Gets***" << std::endl;
testers.push_back(new WaveLevelPrimitiveTester(args));
return testers;
case WAVEGetNBITestType:
if (rank == 0) std::cout << "WAVE Non-Blocking Gets***" << std::endl;
testers.push_back(new WaveLevelPrimitiveTester(args));
return testers;
case WAVEPutTestType:
if (rank == 0) std::cout << "WAVE Blocking Puts***" << std::endl;
testers.push_back(new WaveLevelPrimitiveTester(args));
return testers;
case WAVEPutNBITestType:
if (rank == 0) std::cout << "WAVE Non-Blocking Puts***" << std::endl;
testers.push_back(new WaveLevelPrimitiveTester(args));
return testers;
default:
if (rank == 0) std::cout << "Unknown***" << std::endl;
testers.push_back(new PrimitiveTester(args));