Added logic to clean memory on operations
This commit is contained in:
parent
fbe5947a19
commit
4e2ef636c9
8 changed files with 94 additions and 35 deletions
|
|
@ -30,6 +30,9 @@ class Tensor
|
|||
// Create functions
|
||||
void createBuffer();
|
||||
|
||||
// Destroy/Free functions
|
||||
void freeMemoryDestroyGPUResources();
|
||||
|
||||
// Getter functions
|
||||
std::vector<uint32_t> data();
|
||||
uint32_t size();
|
||||
|
|
@ -54,6 +57,7 @@ class Tensor
|
|||
void mapDataFromHostMemory();
|
||||
void mapDataIntoHostMemory();
|
||||
|
||||
|
||||
private:
|
||||
std::shared_ptr<vk::PhysicalDevice> mPhysicalDevice;
|
||||
std::shared_ptr<vk::Device> mDevice;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue