From 436005cef51967fad73c3f0162a37c95f34f6f0c Mon Sep 17 00:00:00 2001 From: osbm Date: Tue, 10 Dec 2024 20:00:43 +0300 Subject: [PATCH] add main function to invoke GPU memory check --- main.cu | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/main.cu b/main.cu index 92b8891..02c4039 100644 --- a/main.cu +++ b/main.cu @@ -10,4 +10,9 @@ void checkGpuMem() { total_m=(uint)total_t/1048576.0; 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); +} + +int main() { + checkGpuMem(); + return 0; } \ No newline at end of file