#include <stdlib.h>
#include <stdio.h>
#ifdef USE_CUDA
    #ifndef __CUDACC__
        #include "cuda_runtime.h"
    #endif
#endif

#include "include/utils.h"
#include "include/colors.h"


int i_div_up(int a, int b) { // Partie entière supérieure de a/b
    return ((a % b) != 0) ? (a / b + 1) : (a / b);
}

#ifdef __CUDACC__
extern "C" {
#endif
bool check_cuda_compatibility() {
    #ifdef __CUDACC__
    int nDevices;
    cudaDeviceProp prop;

    cudaGetDeviceCount(&nDevices);
    if (nDevices == 0) {
        printf("Pas d'utilisation du GPU\n\n");
        return false;
    }

    printf("GPUs disponibles:\n");

    for (int i=0; i < nDevices; i++) {
        cudaGetDeviceProperties(&prop, i);
        printf(" - %s\n", prop.name);
    }

    cudaGetDeviceProperties(&prop, 0);
    printf("Utilisation du GPU: " BLUE "%s" RESET " (Compute capability: %d.%d)\n\n", prop.name, prop.major, prop.minor);
    return true;
    #else
    printf("Pas d'utilisation du GPU\n\n");
    return false;
    #endif
}
#ifdef __CUDACC__
}
#endif


#ifndef USE_CUDA
    #ifdef __CUDACC__
    extern "C" {
    #endif
    void* nalloc(size_t sz) {
        void* ptr = malloc(sz);
        return ptr;
    }
    #ifdef __CUDACC__
    }
    #endif

    #ifdef __CUDACC__
    extern "C" {
    #endif
    void gree(void* ptr) {
        free(ptr);
    }
    #ifdef __CUDACC__
    }
    #endif
#else
    #ifdef __CUDACC__
    extern "C" {
    #endif
    void* nalloc(size_t sz) {
        void* ptr;
        cudaMallocManaged(&ptr, sz, cudaMemAttachHost);
        return ptr;
    }
    #ifdef __CUDACC__
    }
    #endif

    #ifdef __CUDACC__
    extern "C" {
    #endif
    void gree(void* ptr) {
        cudaFree(ptr);
    }
    #ifdef __CUDACC__
    }
    #endif
#endif