gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix the autotools atrip_cublas

Browse Source
This commit is contained in:
Alejandro Gallo 2022-09-08 13:50:08 +02:00
parent 0b14ac7704
commit 368c5619cc
1 changed files with 23 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
etc/m4/atrip_cublas.m4
View File

@ -40,7 +40,7 @@ AC_RUN_IFELSE([AC_LANG_SOURCE([_ATRIP_CUDA_MEMORY_OF_DEVICES])],
], ],
[ [
atrip_success=no atrip_success=no
AC_MSG_ERROR([An available device reports zero memory available!]) AC_MSG_WARN([An available device reports zero memory available!])
]) ])
CXX="$ac_save_CXX" CXX="$ac_save_CXX"
@ -79,11 +79,11 @@ int main() {
cuMemAlloc((CUdeviceptr*)&F_d, els*sizeof(double)); cuMemAlloc((CUdeviceptr*)&F_d, els*sizeof(double));
stat = cublasDgemm(handle, stat = cublasDgemm(handle,
CUBLAS_OP_N, CUBLAS_OP_N,
CUBLAS_OP_N, CUBLAS_OP_N,
oo, No, Nv, oo, No, Nv,
&one, &one,
HHP_d, oo, PH_d, Nv, &one, F_d, oo); HHP_d, oo, PH_d, Nv, &one, F_d, oo);
//cudaSetDevice(rank); //cudaSetDevice(rank);
return 0; return 0;
@ -92,6 +92,7 @@ int main() {
m4_define([_ATRIP_CUDA_MEMORY_OF_DEVICES], [[ m4_define([_ATRIP_CUDA_MEMORY_OF_DEVICES], [[
#include <mpi.h> #include <mpi.h>
#include <iostream> #include <iostream>
#include <cassert> #include <cassert>
@ -124,26 +125,27 @@ int main() {
cuDeviceTotalMem(&total2, dev); cuDeviceTotalMem(&total2, dev);
printf("\n" printf("\n"
"CUDA CARD RANK %d\n" "CUDA CARD RANK %d\n"
"=================\n" "=================\n"
"\tname: %s\n" "\tname: %s\n"
"\tShared Mem Per Block (KB): %f\n" "\tShared Mem Per Block (KB): %f\n"
"\tFree/Total mem (GB): %f/%f\n" "\tFree/Total mem (GB): %f/%f\n"
"\ttotal2 mem (GB): %f\n" "\ttotal2 mem (GB): %f\n"
"\n", "\n",
dev, dev,
name, name,
prop.sharedMemPerBlock / 1024.0, prop.sharedMemPerBlock / 1024.0,
_free / 1024.0 / 1024.0 / 1024.0 , _free / 1024.0 / 1024.0 / 1024.0 ,
total / 1024.0 / 1024.0 / 1024.0 , total / 1024.0 / 1024.0 / 1024.0 ,
total2 / 1024.0 / 1024.0 / 1024.0 total2 / 1024.0 / 1024.0 / 1024.0
); );
if (_free == 0 || total == 0 || total2 == 0) if (_free == 0 || total == 0 || total2 == 0)
return 1; return 1;
} }
return 0; return 0;
} }
]]) ]])