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

Added initial implementation of shaders as hpp files

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-23 10:56:14 +01:00
parent 2561ace8b7
commit f3b28e6b08
9 changed files with 30 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/OpMult.cpp
View file

@ -1,9 +1,8 @@
#include <chrono>
#include <thread>
#include "Tensor.hpp"
#include "shaders/opmult.hpp"
// Only defining hpp file for syntax validation in editors
#ifndef OPMULT_H
#include "OpMult.hpp"
@ -99,8 +98,10 @@ OpMult<tX, tY, tZ>::init(std::vector<std::shared_ptr<Tensor>> tensors)
this->mDevice,
this->mCommandBuffer);
// TODO: Make this path configurable
this->mAlgorithm->init("shaders/glsl/opmult.comp.spv", tensors);
std::vector<char> shaderFileData(
kp::shader_data::shaders_glsl_opmult_comp_spv_len,
kp::shader_data::shaders_glsl_opmult_comp_spv + kp::shader_data::shaders_glsl_opmult_comp_spv_len);
this->mAlgorithm->init(shaderFileData, tensors);
}
template<uint32_t tX, uint32_t tY, uint32_t tZ>