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

converted roc_shmemx to roc_shmem

Browse Source
This commit is contained in:
Yiltan Temucin
2024-11-12 08:37:56 -06:00
parent 958575d8a4
commit c2b736ef3d
10 changed files with 244 additions and 244 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/functional_tests/extended_primitives.cpp
+4 -4
View File
@@ -54,16 +54,16 @@ __global__ void ExtendedPrimitiveTest(int loop, int skip, uint64_t *timer,
switch (type) {
case WGGetTestType:
roc_shmemx_ctx_getmem_wg(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_getmem_wg(ctx, r_buf, s_buf, size, 1);
break;
case WGGetNBITestType:
roc_shmemx_ctx_getmem_nbi_wg(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_getmem_nbi_wg(ctx, r_buf, s_buf, size, 1);
break;
case WGPutTestType:
roc_shmemx_ctx_putmem_wg(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_putmem_wg(ctx, r_buf, s_buf, size, 1);
break;
case WGPutNBITestType:
roc_shmemx_ctx_putmem_nbi_wg(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_putmem_nbi_wg(ctx, r_buf, s_buf, size, 1);
break;
default:
break;
tests/functional_tests/wave_level_primitives.cpp
+4 -4
View File
@@ -56,16 +56,16 @@ __global__ void WaveLevelPrimitiveTest(int loop, int skip, uint64_t *timer,
switch (type) {
case WAVEGetTestType:
roc_shmemx_ctx_getmem_wave(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_getmem_wave(ctx, r_buf, s_buf, size, 1);
break;
case WAVEGetNBITestType:
roc_shmemx_ctx_getmem_nbi_wave(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_getmem_nbi_wave(ctx, r_buf, s_buf, size, 1);
break;
case WAVEPutTestType:
roc_shmemx_ctx_putmem_wave(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_putmem_wave(ctx, r_buf, s_buf, size, 1);
break;
case WAVEPutNBITestType:
roc_shmemx_ctx_putmem_nbi_wave(ctx, r_buf, s_buf, size, 1);
roc_shmem_ctx_putmem_nbi_wave(ctx, r_buf, s_buf, size, 1);
break;
default:
break;