Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename files to .cu and fix IDX2C usage

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-21 20:03:04 +01:00
parent 15c0cd30f0
commit 035f3b053b
9 changed files with 72 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/layers/dense.h
View File

@@ -1,5 +1,3 @@
// fully_connected_layer.h
#ifndef DENSE_LAYER_H
#define DENSE_LAYER_H

2
include/utils/cuda_helper.h
View File

@@ -24,7 +24,7 @@ do { \
cublasStatus_t result = call; \
if (result != CUBLAS_STATUS_SUCCESS) { \
fprintf(stderr, "cuBLAS error at %s:%d code=%d\n", \
__FILE__, __LINE__, static_cast<int>(result)); \
__FILE__, __LINE__, static_cast<unsigned int>(result)); \
exit(EXIT_FAILURE); \
} \
} while (0)