Added logic to clean memory on operations

This commit is contained in:
Alejandro Saucedo 2020-08-27 06:27:24 +01:00
parent fbe5947a19
commit 4e2ef636c9
8 changed files with 94 additions and 35 deletions

View file

@ -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;