thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Added buffer memory barriers

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-22 13:01:04 +01:00
parent 5bb9046b49
commit 25d5df21ce
4 changed files with 40 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/Tensor.hpp
View file

@ -51,8 +51,8 @@ class Tensor
// Record functions
void recordCopyFrom(std::shared_ptr<Tensor> copyFromTensor);
// TODO: Add memory buffer barrier capabilities
// void recordBufferMemoryBarrier();
// TODO: Explore simplifying by infering pipeline stage flag bits from access flag bits (as seems to be superset)
void recordBufferMemoryBarrier(vk::AccessFlagBits srcAccessMask, vk::AccessFlagBits dstAccessMask, vk::PipelineStageFlagBits srcStageMask, vk::PipelineStageFlagBits dstStageMask);
// Util functions
vk::DescriptorBufferInfo constructDescriptorBufferInfo();