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

Added initial base for opcreatetensor

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-18 21:13:33 +01:00
parent 93041b4519
commit 014f15d552
14 changed files with 422 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/OpCreateTensor.hpp
View file

@ -10,22 +10,31 @@
#include <spdlog/spdlog.h>
#include "Tensor.hpp"
#include "BaseOp.hpp"
namespace kp {
class OpCreateTensor : BaseOp
class OpCreateTensor : BaseOp<OpCreateTensor>
{
private:
public:
OpCreateTensor();
OpCreateTensor(std::shared_ptr<vk::CommandBuffer> commandBuffer);
OpCreateTensor(std::shared_ptr<vk::PhysicalDevice> physicalDevice,
std::shared_ptr<vk::Device> device,
std::shared_ptr<vk::CommandBuffer> commandBuffer);
~OpCreateTensor();
void init(std::shared_ptr<Tensor> tensor, std::vector<uint32_t> data);
void record();
private:
std::shared_ptr<vk::CommandBuffer> mCommandBuffer;
std::shared_ptr<Tensor> mPrimaryTensor;
std::shared_ptr<Tensor> mStagingTensor;
};
} // End namespace kp