Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-07 02:04:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove cublas dependency

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-05 18:41:35 +01:00
parent 98ad84c659
commit f4257afd5a
16 changed files with 65 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/layers/dense.cuh
View File

@@ -1,8 +1,6 @@
#ifndef DENSE_LAYER_H
#define DENSE_LAYER_H
#include <cublas_v2.h>
#include <functional>
#include <string>
#include <vector>
@@ -16,8 +14,7 @@ class Dense : public ILayer {
Dense(
int inputSize,
int outputSize,
Activation activation,
cublasHandle_t cublasHandle
Activation activation
);
~Dense();
@@ -29,8 +26,6 @@ class Dense : public ILayer {
int inputSize;
int outputSize;
cublasHandle_t cublasHandle;
float* d_weights;
float* d_biases;