augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-24 07:36:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update cuda_utils.cu

Browse Source
This commit is contained in:
augustin64 2022-06-27 10:59:29 +02:00
parent 2a0fb52ae4
commit 1f655fecc5
2 changed files with 43 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/mnist/cuda_utils.cu
View File

@ -1,29 +1,45 @@
#include <stdio.h>
#include <stdlib.h>
#include "mnist.c"
#include "include/cuda_utils.h"
int*** copy_images_cuda(int*** images, int nb_images, int width, int height) {
int*** images_cuda;
cudaMalloc(&images_cuda, (size_t)sizeof(int**)*nb_images);
cudaMemcpy(images_cuda, &images, (size_t)sizeof(int**)*nb_images, cudaMemcpyHostToDevice);
for (int i=0; i < nb_images; i++) {
cudaMalloc(&images_cuda[i], sizeof(int**)*nb_images);
cudaMemcpy(images_cuda[i], &images[i], sizeof(int**)*nb_images, cudaMemcpyHostToDevice);
for (int j=0; j < height; j++) {
cudaMalloc((int**)&images_cuda[i][j], sizeof(int*)*width);
cudaMemcpy(images_cuda[i][j], &images[i][j], sizeof(int*)*width, cudaMemcpyHostToDevice);
}
}
return images_cuda;
}
unsigned int* copy_labels_cuda(unsigned int* labels) {
unsigned int* cudaReadMnistLabels(char* filename) {
FILE* ptr;
ptr = fopen(filename, "rb");
uint32_t magic_number;
uint32_t number_of_items;
unsigned int* labels;
unsigned int* labels_cuda;
fread(&magic_number, sizeof(uint32_t), 1, ptr);
magic_number = swap_endian(magic_number);
if (magic_number != 2049) {
printf("Incorrect magic number !\n");
exit(1);
}
fread(&number_of_items, sizeof(uint32_t), 1, ptr);
number_of_items = swap_endian(number_of_items);
unsigned char buffer[number_of_items];
fread(buffer, sizeof(unsigned char), number_of_items, ptr);
labels = (unsigned int*)malloc(sizeof(unsigned int)*number_of_items);
for (int i=0; i < (int)number_of_items; i++) {
labels[i] = (unsigned int)buffer[i];
}
cudaMalloc(&labels_cuda, (size_t)sizeof(labels));
cudaMemcpy(labels_cuda, &labels, sizeof(labels), cudaMemcpyHostToDevice);
free(labels);
return labels_cuda;
}

13
src/mnist/include/cuda_utils.h
View File

@ -3,9 +3,18 @@
#ifndef DEF_CUDA_UTILS_H
#define DEF_CUDA_UTILS_H
/*
* Il est entendu par "device" le GPU supportant CUDA utilisé
*/
int*** copy_images_cuda(int*** images, int nb_images, int width, int height);
unsigned int* copy_labels_cuda(unsigned int* labels);
/*
* Lecture des labels et écriture dans la mémoire du device
*/
unsigned int* cudaReadMnistLabels(char* label_file);
/*
* Vérification de la disponibilité d'un device
*/
void check_cuda_compatibility();
#endif