Fix g/p tests (#266)

[ROCm/rocshmem commit: 6bb46887e8]
This commit is contained in:
Yiltan
2025-09-29 14:27:25 -04:00
committad av GitHub
förälder f18abbc1f2
incheckning 4f955324ac
2 ändrade filer med 9 tillägg och 7 borttagningar
@@ -82,16 +82,14 @@ __global__ void PrimitiveTest(int loop, int skip, long long int *start_time,
rocshmem_ctx_putmem_nbi(ctx, dest, source, size, 1);
break;
case PTestType:
for (int s = 0; s < size; s++) {
char val = source[s];
rocshmem_ctx_char_p(ctx, &dest[s], val, 1);
{
/* Assigment required to verify we can send non-symetric memory */
char val = *source;
rocshmem_ctx_char_p(ctx, dest, val, 1);
}
break;
case GTestType:
for (int s = 0; s < size; s++) {
char ret = rocshmem_ctx_char_g(ctx, &source[s], 1);
dest[s] = ret;
}
*dest = rocshmem_ctx_char_g(ctx, source, 1);
break;
default:
break;
@@ -121,6 +121,10 @@ TesterArguments::TesterArguments(int argc, char *argv[]) {
case PutNBIMRTestType:
min_msg_size = max_msg_size;
break;
case PTestType:
case GTestType:
min_msg_size = 1;
max_msg_size = 1;
default:
break;
}