Brandon Potter
128 строки
3.6 KiB
C++
128 строки
3.6 KiB
C++
/*
|
|
* 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.
|
|
*/
|
|
|
|
/* long_finc neighbor - Perf test roc_shmem_atomic_fetch_inc(); */
|
|
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/time.h>
|
|
|
|
#include <roc_shmem/roc_shmem.hpp>
|
|
|
|
using namespace rocshmem;
|
|
|
|
#define LOOPS 25000
|
|
|
|
static double shmemx_wtime(void) {
|
|
struct timeval tv;
|
|
gettimeofday(&tv, NULL);
|
|
return (double)tv.tv_sec + (double)tv.tv_usec / 1000000.0;
|
|
}
|
|
|
|
int Verbose;
|
|
double elapsed;
|
|
|
|
int main(int argc, char *argv[]) {
|
|
int rc = 0, my_pe, npes, neighbor;
|
|
int loops = LOOPS;
|
|
int j;
|
|
size_t data_sz = sizeof(long) * 3;
|
|
double start_time;
|
|
long *data, lval = 0;
|
|
|
|
if (argc > 1) loops = atoi(argv[1]);
|
|
|
|
roc_shmem_init();
|
|
|
|
my_pe = roc_shmem_my_pe();
|
|
npes = roc_shmem_n_pes();
|
|
|
|
if (loops <= 0) {
|
|
if (my_pe == 0) printf("Error: loops must be greater than 0\n");
|
|
|
|
roc_shmem_finalize();
|
|
return 1;
|
|
}
|
|
|
|
data = (long *)roc_shmem_malloc(data_sz);
|
|
if (!data) {
|
|
fprintf(stderr, "[%d] roc_shmem_malloc(%ld) failure? %d\n", my_pe, data_sz,
|
|
errno);
|
|
roc_shmem_global_exit(1);
|
|
}
|
|
memset((void *)data, 0, data_sz);
|
|
|
|
roc_shmem_barrier_all();
|
|
|
|
neighbor = (my_pe + 1) % npes;
|
|
start_time = shmemx_wtime();
|
|
for (j = 0, elapsed = 0.0; j < loops; j++) {
|
|
start_time = shmemx_wtime();
|
|
lval = roc_shmem_int64_atomic_fetch_inc((int64_t *)&data[1], neighbor);
|
|
elapsed += shmemx_wtime() - start_time;
|
|
if (lval != (long)j) {
|
|
fprintf(stderr, "[%d] Test: FAIL previous val %ld != %d Exit.\n", my_pe,
|
|
lval, j);
|
|
roc_shmem_global_exit(1);
|
|
}
|
|
}
|
|
roc_shmem_barrier_all();
|
|
|
|
rc = 0;
|
|
if (data[1] != (long)loops) {
|
|
fprintf(stderr, "[%d] finc neighbot: FAIL data[1](%p) %ld != %d Exit.\n",
|
|
my_pe, (void *)&data[1], data[1], loops);
|
|
rc--;
|
|
}
|
|
|
|
/* check if adjancent memory locations distrubed */
|
|
assert(data[0] == 0);
|
|
assert(data[2] == 0);
|
|
|
|
if (my_pe == 0) {
|
|
if (rc == 0 && Verbose)
|
|
fprintf(stderr, "[%d] finc neighbor: PASSED.\n", my_pe);
|
|
fprintf(
|
|
stderr,
|
|
"[%d] %d loops of roc_shmem_int64_atomic_fetch_inc() in %6.4f secs\n"
|
|
" %2.6f usecs per roc_shmem_int64_atomic_fetch_inc()\n",
|
|
my_pe, loops, elapsed, ((elapsed * 100000.0) / (double)loops));
|
|
}
|
|
roc_shmem_free(data);
|
|
|
|
roc_shmem_finalize();
|
|
|
|
return rc;
|
|
}
|