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

Updated test

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-20 11:12:42 +00:00
parent 3cee0d9980
commit f713b56862
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
test/TestUtils.cpp
View file

@ -2,6 +2,8 @@
#include <iostream>
#include <vector>
#include "kompute/Kompute.hpp"
#include <glslang/Public/ShaderLang.h>
#include <StandAlone/ResourceLimits.h>
#include <SPIRV/GlslangToSpv.h>
@ -14,6 +16,7 @@ static std::vector<char> spirv_from_sources(const std::vector<std::string>& sour
// Initialize glslang library.
glslang::InitializeProcess();
// Currently we don't support other shader types nor plan to
const EShLanguage language = EShLangCompute;
glslang::TShader shader(language);
@ -49,6 +52,7 @@ static std::vector<char> spirv_from_sources(const std::vector<std::string>& sour
if (!program.link(messages))
{
info_log = std::string(program.getInfoLog()) + "\n" + std::string(program.getInfoDebugLog());
SPDLOG_ERROR(info_log);
throw std::runtime_error(info_log);
}
@ -58,16 +62,12 @@ static std::vector<char> spirv_from_sources(const std::vector<std::string>& sour
info_log += std::string(shader.getInfoLog()) + "\n" + std::string(shader.getInfoDebugLog()) + "\n";
}
if (program.getInfoLog())
{
info_log += std::string(program.getInfoLog()) + "\n" + std::string(program.getInfoDebugLog());
}
glslang::TIntermediate *intermediate = program.getIntermediate(language);
// Translate to SPIRV.
if (!intermediate)
{
info_log += "Failed to get shared intermediate code.\n";
SPDLOG_ERROR(info_log);
throw std::runtime_error(info_log);
}