Remove cublas dependency

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;
 

include/layers/dense.cuh

@@ -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;
 

include/layers/ilayer.cuh

@@ -2,8 +2,6 @@
 #ifndef I_LAYER_H
 #define I_LAYER_H
 
-#include <cublas_v2.h>
-
 #include <vector>
 
 namespace Layers {