mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Implement backend factory
This commit is contained in:
@@ -27,7 +27,14 @@ do { \
|
||||
namespace CUDANet::Backends {
|
||||
|
||||
class CUDA : public Backend {
|
||||
private:
|
||||
int device_id;
|
||||
public:
|
||||
CUDA(const BackendConfig& config);
|
||||
|
||||
static bool is_cuda_available();
|
||||
void initialize();
|
||||
|
||||
// Memory management
|
||||
void* allocate(size_t bytes) override;
|
||||
void deallocate(void* ptr) override;
|
||||
|
||||
Reference in New Issue
Block a user