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