Remove cublas dependency

2025-11-06 01:34:22 +00:00 · 2024-03-05 18:41:35 +01:00
parent 98ad84c659
commit f4257afd5a
16 changed files with 65 additions and 141 deletions
--- a/include/layers/conv2d.cuh
+++ b/include/layers/conv2d.cuh
@@ -1,8 +1,6 @@
 #ifndef CONV_LAYER_H
 #define CONV_LAYER_H

-#include <cublas_v2.h>
-
 #include <string>
 #include <vector>

@@ -19,8 +17,7 @@ class Conv2d {
        int            stride,
        std::string    padding,
        int            numFilters,
-        Activation     activation,
-        cublasHandle_t cublasHandle
+        Activation     activation
    );
    ~Conv2d();

@@ -44,7 +41,6 @@ class Conv2d {
    std::vector<float> kernels;

    // Cuda
-    cublasHandle_t cublasHandle;
    float*         d_kernels;
    float*         d_padded;