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

Transfer files from RAD repository

Browse Source
This commit is contained in:
Brandon Potter
2024-07-01 09:57:08 -05:00
parent a30da178c1
commit ea8f264a11
382 changed files with 67034 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/sos_tests/bcast_flood.cpp
+222
View File
@@ -0,0 +1,222 @@
/*
* Copyright 2011 Sandia Corporation. Under the terms of Contract
* DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government
* retains certain rights in this software.
*
* Copyright (c) 2017 Intel Corporation. All rights reserved.
* This software is available to you under the BSD license below:
*
* Redistribution and use in source and binary forms, with or
* without modification, are permitted provided that the following
* conditions are met:
*
* - Redistributions of source code must retain the above
* copyright notice, this list of conditions and the following
* disclaimer.
*
* - Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials
* provided with the distribution.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
/* Broadcast flood - bcast 100KB to all, default to using 2 pSync vars */
#include <assert.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/time.h>
#include <roc_shmem/roc_shmem.hpp>
using namespace rocshmem;
static int atoi_scaled(char *s);
static void usage(char *pgm);
static double shmemx_wtime(void) {
struct timeval tv;
gettimeofday(&tv, NULL);
return (double)tv.tv_sec + (double)tv.tv_usec / 1000000.0;
}
int Verbose = 0;
int Serialize;
long *pSync;
#define DFLT_LOOPS 600 // downsized for 'make check'
//#define DFLT_LOOPS 10000
#define N_ELEMENTS 25600 /*100 KB as ints */
int main(int argc, char **argv) {
int i, ps, ps_cnt = 2;
int *target;
int *source;
int me, npes, elements = N_ELEMENTS, loops = DFLT_LOOPS;
char *pgm;
double start_time, time_taken;
roc_shmem_init();
me = roc_shmem_my_pe();
npes = roc_shmem_n_pes();
if ((pgm = strrchr(argv[0], '/'))) {
pgm++;
} else {
pgm = argv[0];
}
while ((i = getopt(argc, argv, "hve:l:p:s")) != EOF) {
switch (i) {
case 'v':
Verbose++;
break;
case 'e':
if ((elements = atoi_scaled(optarg)) <= 0) {
fprintf(stderr, "ERR: Bad elements count %d\n", elements);
roc_shmem_finalize();
return 1;
}
break;
case 'l':
if ((loops = atoi_scaled(optarg)) <= 0) {
fprintf(stderr, "ERR: Bad loop count %d\n", loops);
roc_shmem_finalize();
return 1;
}
break;
case 'p':
if ((ps_cnt = atoi_scaled(optarg)) <= 0) {
fprintf(stderr, "ERR: Bad pSync[] elements %d\n", loops);
roc_shmem_finalize();
return 1;
}
break;
case 's':
Serialize++;
break;
case 'h':
if (me == 0) usage(pgm);
return 0;
default:
if (me == 0) {
fprintf(stderr, "%s: unknown switch '-%c'?\n", pgm, i);
usage(pgm);
}
roc_shmem_finalize();
return 1;
}
}
ps_cnt *= ROC_SHMEM_BCAST_SYNC_SIZE;
pSync = (long *)roc_shmem_malloc(ps_cnt * sizeof(long));
if (!pSync) {
fprintf(stderr, "ERR - null pSync pointer\n");
roc_shmem_global_exit(1);
}
for (i = 0; i < ps_cnt; i++) {
pSync[i] = ROC_SHMEM_SYNC_VALUE;
}
source = (int *)roc_shmem_malloc(elements * sizeof(*source));
if (!source) {
fprintf(stderr, "ERR - null source pointer\n");
roc_shmem_global_exit(1);
}
target = (int *)roc_shmem_malloc(elements * sizeof(*target));
if (!target) {
fprintf(stderr, "ERR - null target pointer\n");
roc_shmem_global_exit(1);
}
for (i = 0; i < elements; i += 1) {
source[i] = i + 1;
target[i] = -90;
}
if (me == 0 && Verbose) {
fprintf(stderr, "ps_cnt %d loops %d nElems %d\n", ps_cnt, loops, elements);
}
roc_shmem_barrier_all();
for (time_taken = 0.0, ps = i = 0; i < loops; i++) {
start_time = shmemx_wtime();
roc_shmem_ctx_int_broadcast(ROC_SHMEM_CTX_DEFAULT, target, source, elements,
0, 0, 0, npes, &pSync[ps]);
if (Serialize) roc_shmem_barrier_all();
time_taken += (shmemx_wtime() - start_time);
if (ps_cnt > 1) {
ps += ROC_SHMEM_BCAST_SYNC_SIZE;
if (ps >= ps_cnt) ps = 0;
}
}
if (me == 0 && Verbose) {
printf("%d loops of int_broadcast(%ld bytes) over %d PEs: %7.3f secs\n",
loops, (elements * sizeof(*source)), npes, time_taken);
elements = (elements * loops * sizeof(*source)) / (1024 * 1024);
printf(" %7.5f secs per broadcast() @ %7.4f MB/sec\n",
(time_taken / (double)loops), ((double)elements / time_taken));
}
if (Verbose > 1) fprintf(stderr, "[%d] pre B1\n", me);
roc_shmem_barrier_all();
if (Verbose > 1) fprintf(stderr, "[%d] post B1\n", me);
roc_shmem_free(pSync);
roc_shmem_free(target);
roc_shmem_free(source);
roc_shmem_finalize();
return 0;
}
static int atoi_scaled(char *s) {
long val;
char *e;
val = strtol(s, &e, 0);
if (e == NULL || *e == '\0') return (int)val;
if (*e == 'k' || *e == 'K')
val *= 1024;
else if (*e == 'm' || *e == 'M')
val *= 1024 * 1024;
else if (*e == 'g' || *e == 'G')
val *= 1024 * 1024 * 1024;
return (int)val;
}
static void usage(char *pgm) {
fprintf(stderr,
"usage: %s -{lhv}\n"
" where:\n"
" -l loops (%d) loop count.\n"
" -e ints # of integers to broadcast\n"
" -p cnt # of pSync[] elements\n"
" -v be verbose, multiple 'v' more verbose\n"
" -h this text.\n",
pgm, DFLT_LOOPS);
}