mirror of
https://github.com/augustin64/projet-tipe
synced 2025-01-23 23:26:25 +01:00
42 lines
1.1 KiB
Plaintext
42 lines
1.1 KiB
Plaintext
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "../src/mnist/include/cuda_utils.h"
|
|
#define MAX_CUDA_THREADS 1024
|
|
|
|
#define gpuErrchk(ans) { gpuAssert((ans), __FILE__, __LINE__); }
|
|
inline void gpuAssert(cudaError_t code, const char *file, int line, bool abort=true)
|
|
{
|
|
if (code != cudaSuccess)
|
|
{
|
|
fprintf(stderr,"GPUassert: %s %s %d\n", cudaGetErrorString(code), file, line);
|
|
if (abort) exit(code);
|
|
}
|
|
}
|
|
|
|
__global__ void check_labels(int n, unsigned int* labels) {
|
|
for (int i=0; i < n; i++) {
|
|
(void)labels[i];
|
|
}
|
|
}
|
|
|
|
|
|
int main() {
|
|
printf("Test de la compatibilité CUDA\n");
|
|
check_cuda_compatibility();
|
|
printf("OK\n");
|
|
|
|
printf("Lecture des labels\n");
|
|
unsigned int* labels = cudaReadMnistLabels("data/mnist/t10k-labels-idx1-ubyte");
|
|
printf("OK\n");
|
|
|
|
printf("Test des labels\n");
|
|
//! TODO: fix
|
|
// Ne provoque pas d'erreurs, mais tous les labels valent 1
|
|
check_labels<<<1, 1>>>(10000, labels);
|
|
gpuErrchk( cudaPeekAtLastError() );
|
|
gpuErrchk( cudaDeviceSynchronize() );
|
|
printf("OK\n");
|
|
|
|
return 0;
|
|
} |