Added initial implementation for algorithm and opMult
This commit is contained in:
parent
0d18dc50e6
commit
d59dc41ffc
11 changed files with 385 additions and 28 deletions
|
|
@ -46,8 +46,18 @@ class Tensor
|
|||
TensorTypes tensorType();
|
||||
bool isInit();
|
||||
|
||||
// Setters
|
||||
void setData(const std::vector<uint32_t>& data);
|
||||
|
||||
// Record functions
|
||||
void recordCopyFrom(std::shared_ptr<Tensor> copyFromTensor);
|
||||
// TODO: Add memory buffer barrier capabilities
|
||||
//void recordBufferMemoryBarrier();
|
||||
|
||||
// Util functions
|
||||
vk::DescriptorBufferInfo constructDescriptorBufferInfo();
|
||||
void copyDataFromHostBuffer();
|
||||
void copyDataToHostBuffer();
|
||||
|
||||
private:
|
||||
std::shared_ptr<vk::PhysicalDevice> mPhysicalDevice;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue