mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 22:34:22 +00:00
Fix compilation errors and warnings
This commit is contained in:
@@ -8,9 +8,10 @@
|
||||
|
||||
namespace CUDANet {
|
||||
|
||||
// Forward declaration
|
||||
class Tensor;
|
||||
// Forward declarations
|
||||
class Backend;
|
||||
class Tensor;
|
||||
enum class DType;
|
||||
|
||||
enum BackendType { CUDA_BACKEND, CPU_BACKEND };
|
||||
|
||||
|
||||
@@ -29,6 +29,8 @@ size_t dtype_size(DType dtype) {
|
||||
}
|
||||
}
|
||||
|
||||
class Backend;
|
||||
|
||||
class Tensor
|
||||
{
|
||||
public:
|
||||
@@ -49,6 +51,7 @@ public:
|
||||
size_t size() const;
|
||||
size_t numel() const;
|
||||
|
||||
void* device_ptr() const;
|
||||
void* device_ptr();
|
||||
|
||||
void zero();
|
||||
|
||||
Reference in New Issue
Block a user