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

Introduced simplified interface for creation of algobase

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-31 09:35:40 +01:00
parent 13206a9d9b
commit f29e56fcb8
4 changed files with 173 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

8
test/TestCustomOpRawShaders.cpp
View file

@ -10,7 +10,7 @@ TEST_CASE("op_custom_simple_raw_shader") {
std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor({ 3, 4, 5 })};
std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor({ 0, 0, 0 })};
mgr.evalOp<kp::OpCreateTensor>({ tensorA, tensorB });
mgr.evalOpDefault<kp::OpCreateTensor>({ tensorA, tensorB });
std::string shader(
"#version 450\n"
@ -24,11 +24,9 @@ TEST_CASE("op_custom_simple_raw_shader") {
"}\n"
);
mgr.evalOp<kp::OpAlgoBase<>>(
mgr.evalOpDefault<kp::OpAlgoBase<>>(
{ tensorA, tensorB },
"DEFAULT",
true,
"",
true, // Whether to copy output from device
std::vector<char>(shader.begin(), shader.end()));
REQUIRE(tensorA->data() == std::vector<uint32_t>{0, 1, 2});