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

Add alltoall perf test

Browse Source
This commit is contained in:
Sylvain Jeaugey
2020-03-17 12:00:19 -07:00
parent cbe7f65400
commit 119a0ecf60
4 changed files with 129 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/common.h
+8 -6
View File
@@ -17,25 +17,25 @@
#include "nccl1_compat.h"
#define CUDACHECK(cmd) do { \
cudaError_t e = cmd; \
if( e != cudaSuccess ) { \
cudaError_t err = cmd; \
if( err != cudaSuccess ) { \
char hostname[1024]; \
getHostName(hostname, 1024); \
printf("%s: Test CUDA failure %s:%d '%s'\n", \
hostname, \
__FILE__,__LINE__,cudaGetErrorString(e)); \
__FILE__,__LINE__,cudaGetErrorString(err)); \
return testCudaError; \
} \
} while(0)
#define NCCLCHECK(cmd) do { \
ncclResult_t r = cmd; \
if (r!= ncclSuccess) { \
ncclResult_t res = cmd; \
if (res != ncclSuccess) { \
char hostname[1024]; \
getHostName(hostname, 1024); \
printf("%s: Test NCCL failure %s:%d '%s'\n", \
hostname, \
__FILE__,__LINE__,ncclGetErrorString(r)); \
__FILE__,__LINE__,ncclGetErrorString(res)); \
return testNcclError; \
} \
} while(0)
@@ -124,6 +124,8 @@ struct threadArgs {
double* bw;
int* bw_count;
int reportErrors;
struct testColl* collTest;
};