Implement to_cuda function

This commit is contained in:
2024-02-09 19:18:30 +01:00
parent 6645cd31ba
commit 3e6b7dc7e6
4 changed files with 37 additions and 6 deletions

19
include/layers/ilayer.h Normal file
View File

@@ -0,0 +1,19 @@
#ifndef I_LAYER_H
#define I_LAYER_H
#include <cublas_v2.h>
namespace Layers {
class ILayer {
public:
virtual ~ILayer() {}
virtual void forward(const float* input, float* output) = 0;
virtual void to_cuda() = 0;
};
} // namespace Layers
#endif // I_LAYERH