From 57954a27c09130fdb621bf0ffdbfab28096e6379 Mon Sep 17 00:00:00 2001
From: augustin64 <me.git@augustin64.fr>
Date: Mon, 27 Mar 2023 18:30:19 +0200
Subject: [PATCH] Update tests

---
 test/cnn_function.cu      | 11 ++++++++++-
 test/memory_management.cu |  8 ++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/test/cnn_function.cu b/test/cnn_function.cu
index 931a50b..54b7fd3 100644
--- a/test/cnn_function.cu
+++ b/test/cnn_function.cu
@@ -5,10 +5,19 @@
 #include "../src/include/memory_management.h"
 #include "../src/cnn/include/function.h"
 #include "../src/include/colors.h"
+#include "../src/include/utils.h"
 
 
 int main() {
-    printf("Initialisation\n");
+    printf("Checking CUDA compatibility.\n");
+    bool cuda_compatible = check_cuda_compatibility();
+    if (!cuda_compatible) {
+        printf(RED "CUDA not compatible, skipping tests.\n" RESET);
+        return 0;
+    }
+    printf(GREEN "OK\n" RESET);
+
+    printf("Initialisation OK\n");
     // Initialise values
     int depth = 10;
     int rows = 10;
diff --git a/test/memory_management.cu b/test/memory_management.cu
index 62df868..62de138 100644
--- a/test/memory_management.cu
+++ b/test/memory_management.cu
@@ -17,6 +17,14 @@ __global__ void check_access(int* array, int range) {
 
 
 int main() {
+    printf("Checking CUDA compatibility.\n");
+    bool cuda_compatible = check_cuda_compatibility();
+    if (!cuda_compatible) {
+        printf(RED "CUDA not compatible, skipping tests.\n" RESET);
+        return 0;
+    }
+    printf(GREEN "OK\n" RESET);
+
     printf("Pollution de la mémoire\n");
     int mem_used;
     int blocks_used;