osbm/cuda-kernel-nix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add main function to invoke GPU memory check

Browse source
This commit is contained in:
Osman Faruk Bayram 2024-12-10 20:00:43 +03:00
parent 728d8c34c0
commit 436005cef5
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
main.cu
View file

@ -11,3 +11,8 @@ void checkGpuMem() {
used_m=total_m-free_m; used_m=total_m-free_m;
printf("GPU memory usage: used = %f, free = %f MB, total = %f MB\n",used_m,free_m,total_m); printf("GPU memory usage: used = %f, free = %f MB, total = %f MB\n",used_m,free_m,total_m);
} }
int main() {
checkGpuMem();
return 0;
}