Remove cublas dependency

This commit is contained in:
2024-03-05 18:41:35 +01:00
parent 98ad84c659
commit f4257afd5a
16 changed files with 65 additions and 141 deletions

View File

@@ -1,5 +1,3 @@
#include <cublas_v2.h>
#include <string>
#include "activations.cuh"
@@ -14,15 +12,13 @@ Layers::Conv2d::Conv2d(
int stride,
std::string padding,
int numFilters,
Activation activation,
cublasHandle_t cublasHandle
Activation activation
)
: inputSize(inputSize),
inputChannels(inputChannels),
kernelSize(kernelSize),
stride(stride),
numFilters(numFilters),
cublasHandle(cublasHandle),
activation(activation) {
// Allocate memory for kernels