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

reformat

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-08 21:42:32 +00:00
parent 71cde2d5b2
commit 3547810476
2 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/include/kompute/Manager.hpp
View file

@ -248,11 +248,10 @@ class Manager
tensor->init(this->mPhysicalDevice, this->mDevice);
if (syncDataToGPU) {
this->evalOpDefault<OpTensorSyncDevice>({tensor});
this->evalOpDefault<OpTensorSyncDevice>({ tensor });
}
this->mManagedTensors.insert(tensor);
return tensor;
}
@ -267,7 +266,8 @@ class Manager
* @param syncDataToGPU Whether to sync the data to GPU memory
* @returns Initialized Tensor with memory Syncd to GPU device
*/
void rebuildTensors(std::vector<std::shared_ptr<kp::Tensor>> tensors, bool syncDataToGPU = true)
void rebuildTensors(std::vector<std::shared_ptr<kp::Tensor>> tensors,
bool syncDataToGPU = true)
{
SPDLOG_DEBUG("Kompute Manager rebuildTensors triggered");
for (std::shared_ptr<Tensor> tensor : tensors) {