augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-23 15:16:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

utils:cuda: Select best available GPU

Browse Source
This commit is contained in:
augustin64 2023-05-16 13:21:26 +02:00
parent e1617a72a8
commit dbd5362d7d
9 changed files with 91 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cnn/train.c
View File

@ -127,7 +127,7 @@ void* train_thread(void* parameters) {
void train(int dataset_type, char* images_file, char* labels_file, char* data_dir, int epochs, char* out, char* recover) {
#ifdef USE_CUDA
bool compatibility = check_cuda_compatibility();
bool compatibility = cuda_setup(true);
if (!compatibility) {
printf("Exiting.\n");
exit(1);

4
src/common/include/utils.h
View File

@ -60,8 +60,10 @@ extern "C"
#endif
/*
* Vérification de la compatibilité CUDA
* spécifier avec "verbose" si il faut afficher
* la carte utilisée notamment
*/
bool check_cuda_compatibility();
bool cuda_setup(bool verbose);
#ifdef __CUDACC__
extern "C"

55
src/common/utils.c
View File

@ -1,6 +1,7 @@
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#ifdef USE_CUDA
#ifndef __CUDACC__
#include "cuda_runtime.h"
@ -42,39 +43,65 @@ int i_div_up(int a, int b) { // Partie entière supérieure de a/b
#ifdef __CUDACC__
extern "C"
#endif
bool check_cuda_compatibility() {
bool cuda_setup(bool verbose) {
#ifdef __CUDACC__
int nDevices;
int selected_device = 0;
cudaDeviceProp selected_prop;
cudaDeviceProp prop;
cudaGetDeviceCount(&nDevices);
if (nDevices == 0) {
printf("Pas d'utilisation du GPU\n\n");
if (nDevices <= 0) { // I've seen weird issues when there is no GPU at all
if (verbose) {
printf("Pas d'utilisation du GPU\n\n");
}
return false;
}
printf("GPUs disponibles:\n");
if (verbose) {
printf("GPUs disponibles:\n");
}
cudaGetDeviceProperties(&selected_prop, selected_device);
for (int i=0; i < nDevices; i++) {
cudaGetDeviceProperties(&prop, i);
printf(" - %s\n\t - Compute Capability: %d.%d\n\t - Memory available: ", prop.name, prop.major, prop.minor);
printf_memory(prop.totalGlobalMem);
printf("\n\t - Shared Memory per block: ");
printf_memory(prop.sharedMemPerBlock);
printf("\n\n");
if (verbose) {
printf(" - %s\n\t - Compute Capability: %d.%d\n\t - Memory available: ", prop.name, prop.major, prop.minor);
printf_memory(prop.totalGlobalMem);
printf("\n\t - Shared Memory per block: ");
printf_memory(prop.sharedMemPerBlock);
printf("\n\n");
}
if (prop.clockRate*prop.multiProcessorCount >= selected_prop.clockRate*selected_prop.multiProcessorCount) { // This criteria approximately matches the best device
selected_prop = prop;
selected_device = i;
}
}
cudaGetDeviceProperties(&prop, 0);
printf("Utilisation du GPU: " BLUE "%s" RESET "\n\n", prop.name);
cudaSetDevice(selected_device); // Select the best device for computation
if (verbose) {
printf("Utilisation du GPU: " BLUE "%s" RESET "\n\n", selected_prop.name);
}
if (prop.sharedMemPerBlock != MEMORY_BLOCK) {
if (BLOCKSIZE_x*BLOCKSIZE_y*BLOCKSIZE_z > prop.maxThreadsPerBlock) {
printf_error((char*)"La taille de bloc sélectionnée est trop grande.\n");
printf("\tMaximum accepté: %d\n", selected_prop.maxThreadsPerBlock);
exit(1);
}
if (selected_prop.sharedMemPerBlock != MEMORY_BLOCK) { // C'est un warning, on l'affiche dans tous les cas
printf_warning((char*)"La taille des blocs mémoire du GPU et celle utilisée dans le code diffèrent.\n");
printf("\tCela peut mener à une utilisation supplémentaire de VRAM.\n");
printf("\tChanger MEMORY_BLOCK à %ld dans src/include/memory_management.h\n", prop.sharedMemPerBlock);
printf("\tChanger MEMORY_BLOCK à %ld dans src/include/memory_management.h\n", selected_prop.sharedMemPerBlock);
}
return true;
#else
printf("Pas d'utilisation du GPU\n\n");
if (verbose) {
printf("Pas d'utilisation du GPU\n\n");
}
return false;
#endif
}

55
src/common/utils.cu
View File

@ -1,6 +1,7 @@
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#ifdef USE_CUDA
#ifndef __CUDACC__
#include "cuda_runtime.h"
@ -42,39 +43,65 @@ int i_div_up(int a, int b) { // Partie entière supérieure de a/b
#ifdef __CUDACC__
extern "C"
#endif
bool check_cuda_compatibility() {
bool cuda_setup(bool verbose) {
#ifdef __CUDACC__
int nDevices;
int selected_device = 0;
cudaDeviceProp selected_prop;
cudaDeviceProp prop;
cudaGetDeviceCount(&nDevices);
if (nDevices == 0) {
printf("Pas d'utilisation du GPU\n\n");
if (nDevices <= 0) { // I've seen weird issues when there is no GPU at all
if (verbose) {
printf("Pas d'utilisation du GPU\n\n");
}
return false;
}
printf("GPUs disponibles:\n");
if (verbose) {
printf("GPUs disponibles:\n");
}
cudaGetDeviceProperties(&selected_prop, selected_device);
for (int i=0; i < nDevices; i++) {
cudaGetDeviceProperties(&prop, i);
printf(" - %s\n\t - Compute Capability: %d.%d\n\t - Memory available: ", prop.name, prop.major, prop.minor);
printf_memory(prop.totalGlobalMem);
printf("\n\t - Shared Memory per block: ");
printf_memory(prop.sharedMemPerBlock);
printf("\n\n");
if (verbose) {
printf(" - %s\n\t - Compute Capability: %d.%d\n\t - Memory available: ", prop.name, prop.major, prop.minor);
printf_memory(prop.totalGlobalMem);
printf("\n\t - Shared Memory per block: ");
printf_memory(prop.sharedMemPerBlock);
printf("\n\n");
}
if (prop.clockRate*prop.multiProcessorCount >= selected_prop.clockRate*selected_prop.multiProcessorCount) { // This criteria approximately matches the best device
selected_prop = prop;
selected_device = i;
}
}
cudaGetDeviceProperties(&prop, 0);
printf("Utilisation du GPU: " BLUE "%s" RESET "\n\n", prop.name);
cudaSetDevice(selected_device); // Select the best device for computation
if (verbose) {
printf("Utilisation du GPU: " BLUE "%s" RESET "\n\n", selected_prop.name);
}
if (prop.sharedMemPerBlock != MEMORY_BLOCK) {
if (BLOCKSIZE_x*BLOCKSIZE_y*BLOCKSIZE_z > prop.maxThreadsPerBlock) {
printf_error((char*)"La taille de bloc sélectionnée est trop grande.\n");
printf("\tMaximum accepté: %d\n", selected_prop.maxThreadsPerBlock);
exit(1);
}
if (selected_prop.sharedMemPerBlock != MEMORY_BLOCK) { // C'est un warning, on l'affiche dans tous les cas
printf_warning((char*)"La taille des blocs mémoire du GPU et celle utilisée dans le code diffèrent.\n");
printf("\tCela peut mener à une utilisation supplémentaire de VRAM.\n");
printf("\tChanger MEMORY_BLOCK à %ld dans src/include/memory_management.h\n", prop.sharedMemPerBlock);
printf("\tChanger MEMORY_BLOCK à %ld dans src/include/memory_management.h\n", selected_prop.sharedMemPerBlock);
}
return true;
#else
printf("Pas d'utilisation du GPU\n\n");
if (verbose) {
printf("Pas d'utilisation du GPU\n\n");
}
return false;
#endif
}

2
src/scripts/convolution_benchmark.cu
View File

@ -217,7 +217,7 @@ int main(int argc, char* argv[]) {
/*
printf("Checking CUDA compatibility.\n");
bool cuda_compatible = check_cuda_compatibility();
bool cuda_compatible = cuda_setup(true);
if (!cuda_compatible) {
printf(RED "CUDA not compatible, skipping tests.\n" RESET);
return 0;

2
test/cnn_convolution.cu
View File

@ -192,7 +192,7 @@ void run_convolution_test(int input_width, int output_width, int rows, int colum
int main() {
printf("Checking CUDA compatibility.\n");
bool cuda_compatible = check_cuda_compatibility();
bool cuda_compatible = cuda_setup(true);
if (!cuda_compatible) {
printf(RED "CUDA not compatible, skipping tests.\n" RESET);
return 0;

2
test/cnn_function.cu
View File

@ -91,7 +91,7 @@ void test1(int activation, bool use_local_kernel) {
int main() {
printf("Checking CUDA compatibility.\n");
bool cuda_compatible = check_cuda_compatibility();
bool cuda_compatible = cuda_setup(true);
if (!cuda_compatible) {
printf(RED "CUDA not compatible, skipping tests.\n" RESET);
return 0;

2
test/cnn_matrix_multiplication.cu
View File

@ -127,7 +127,7 @@ void run_matrices_test(int n, int p, int q) {
int main() {
printf("Checking CUDA compatibility.\n");
bool cuda_compatible = check_cuda_compatibility();
bool cuda_compatible = cuda_setup(true);
if (!cuda_compatible) {
printf(RED "CUDA not compatible, skipping tests.\n" RESET);
return 0;

2
test/memory_management.cu
View File

@ -18,7 +18,7 @@ __global__ void check_access(int* array, int range) {
int main() {
printf("Checking CUDA compatibility.\n");
bool cuda_compatible = check_cuda_compatibility();
bool cuda_compatible = cuda_setup(true);
if (!cuda_compatible) {
printf(RED "CUDA not compatible, skipping tests.\n" RESET);
return 0;