Initialize and use cuBLAS properly

2025-11-06 01:34:22 +00:00 · 2024-02-10 17:17:07 +01:00
parent 3e6b7dc7e6
commit 22e33a395b
5 changed files with 35 additions and 12 deletions
--- a/include/layers/ilayer.h
+++ b/include/layers/ilayer.h
@@ -11,7 +11,6 @@ namespace Layers {
        virtual ~ILayer() {}

        virtual void forward(const float* input, float* output) = 0;
-        virtual void to_cuda() = 0;
    };

 } // namespace Layers