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

Updated issue in opalgolhs

Browse source
This commit is contained in:
Alejandro Saucedo 2020-09-19 16:10:39 +01:00
parent 71a0b24e42
commit c1b9436b8c
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
single_include/kompute/Kompute.hpp
View file

@ -1464,7 +1464,10 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
vk::PipelineStageFlagBits::eComputeShader,
vk::PipelineStageFlagBits::eTransfer);
this->mTensorOutputStaging->recordCopyFrom(this->mTensorOutput, true);
this->mTensorOutputStaging->recordCopyFrom(
this->mCommandBuffer,
this->mTensorOutput,
true);
}
template<uint32_t tX, uint32_t tY, uint32_t tZ>

5
src/include/kompute/operations/OpAlgoLhsRhsOut.hpp
View file

@ -199,7 +199,10 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
vk::PipelineStageFlagBits::eComputeShader,
vk::PipelineStageFlagBits::eTransfer);
this->mTensorOutputStaging->recordCopyFrom(this->mTensorOutput, true);
this->mTensorOutputStaging->recordCopyFrom(
this->mCommandBuffer,
this->mTensorOutput,
true);
}
template<uint32_t tX, uint32_t tY, uint32_t tZ>