57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
#include "gtest/gtest.h"
|
|
|
|
#include "kompute/Kompute.hpp"
|
|
|
|
static const std::string shaderString = (R"(
|
|
#version 450
|
|
|
|
layout (local_size_x = 1) in;
|
|
|
|
// The input tensors bind index is relative to index in parameter passed
|
|
layout(set = 0, binding = 0) buffer bina { float tina[]; };
|
|
layout(set = 0, binding = 1) buffer binb { float tinb[]; };
|
|
layout(set = 0, binding = 2) buffer bout { float tout[]; };
|
|
|
|
void main() {
|
|
uint index = gl_GlobalInvocationID.x;
|
|
|
|
int i = 1;
|
|
|
|
while(i < 200) {
|
|
tout[index] += tina[index] * tinb[index];
|
|
i++;
|
|
}
|
|
}
|
|
)");
|
|
|
|
void compileShaderWithGivenResources(const std::string shaderString, const TBuiltInResource resources) {
|
|
kp::Shader::compile_source(shaderString, std::string("main"), std::vector<std::pair<std::string,std::string>>({}), resources);
|
|
}
|
|
|
|
|
|
|
|
TEST(TestShaderResources, TestNoMaxLight)
|
|
{
|
|
TBuiltInResource noMaxLightResources = kp::Shader::defaultResource;
|
|
noMaxLightResources.maxLights=0;
|
|
|
|
EXPECT_NO_THROW(compileShaderWithGivenResources(shaderString, noMaxLightResources));
|
|
}
|
|
|
|
|
|
TEST(TestShaderResources, TestSmallComputeWorkGroupSizeX)
|
|
{
|
|
TBuiltInResource smallComputeWorkGroupSizeXResources = kp::Shader::defaultResource;
|
|
smallComputeWorkGroupSizeXResources.maxComputeWorkGroupSizeX=0;
|
|
|
|
ASSERT_THROW(compileShaderWithGivenResources(shaderString, smallComputeWorkGroupSizeXResources), std::runtime_error);
|
|
}
|
|
|
|
|
|
TEST(TestShaderResources, TestNoWhileLoopLimit)
|
|
{
|
|
TBuiltInResource noWhileLoopLimitResources = kp::Shader::defaultResource;
|
|
noWhileLoopLimitResources.limits.whileLoops=0;
|
|
|
|
ASSERT_THROW(compileShaderWithGivenResources(shaderString, noWhileLoopLimitResources), std::runtime_error);
|
|
}
|