From 349ec669ad74c7449a52e676c32a31e25a1faa29 Mon Sep 17 00:00:00 2001
From: osbm <osmanfbayram@gmail.com>
Date: Tue, 10 Dec 2024 20:47:16 +0300
Subject: [PATCH] updated main

---
 main.cu | 25 ++++++++++---------------
 1 file changed, 10 insertions(+), 15 deletions(-)

diff --git a/main.cu b/main.cu
index 02c4039..ae758a0 100644
--- a/main.cu
+++ b/main.cu
@@ -1,18 +1,13 @@
-#include <stdio.h>
-#include "cuda.h"
-extern "C"
+#include <unistd.h>
+#include <iostream>
 
-void checkGpuMem() {
-    float free_m, total_m, used_m;
-    size_t free_t,total_t;
-    cudaMemGetInfo(&free_t,&total_t);
-    free_m =(uint)free_t/1048576.0 ;
-    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();
+int main(){
+
+    size_t mf, ma;
+    cudaMemGetInfo(&mf, &ma);
+    std::cout << "Free memory (gb): " << mf/1024/1024/1024 << std::endl;
+    std::cout << "Total memory (gb): " << ma/1024/1024/1024 << std::endl;
+
     return 0;
-}
\ No newline at end of file
+}