1 /*------------------------------------------------------------------------
2 * OpenGL Conformance Tests
3 * ------------------------
5 * Copyright (c) 2017-2019 The Khronos Group Inc.
6 * Copyright (c) 2017 Codeplay Software Ltd.
7 * Copyright (c) 2019 NVIDIA Corporation.
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
23 * \brief Subgroups Tests
24 */ /*--------------------------------------------------------------------*/
26 #include "glcSubgroupsBallotBroadcastTests.hpp"
27 #include "glcSubgroupsTestsUtils.hpp"
44 OPTYPE_BROADCAST_FIRST,
48 static bool checkVertexPipelineStages(std::vector<const void*> datas,
49 deUint32 width, deUint32)
51 return glc::subgroups::check(datas, width, 3);
54 static bool checkComputeStages(std::vector<const void*> datas,
55 const deUint32 numWorkgroups[3], const deUint32 localSize[3],
58 return glc::subgroups::checkCompute(datas, numWorkgroups, localSize, 3);
61 std::string getOpTypeName(int opType)
66 DE_FATAL("Unsupported op type");
68 case OPTYPE_BROADCAST:
69 return "subgroupBroadcast";
70 case OPTYPE_BROADCAST_FIRST:
71 return "subgroupBroadcastFirst";
78 ShaderStageFlags shaderStage;
82 std::string getBodySource(CaseDefinition caseDef)
84 std::ostringstream bdy;
86 bdy << " uvec4 mask = subgroupBallot(true);\n";
87 bdy << " uint tempResult = 0;\n";
89 if (OPTYPE_BROADCAST == caseDef.opType)
91 bdy << " tempResult = 0x3;\n";
92 for (int i = 0; i < (int)subgroups::maxSupportedSubgroupSize(); i++)
95 << " const uint id = "<< i << ";\n"
96 << " " << subgroups::getFormatNameForGLSL(caseDef.format)
97 << " op = subgroupBroadcast(data1[gl_SubgroupInvocationID], id);\n"
98 << " if ((id < gl_SubgroupSize) && subgroupBallotBitExtract(mask, id))\n"
100 << " if (op != data1[id])\n"
102 << " tempResult = 0;\n"
110 bdy << " uint firstActive = 0;\n"
111 << " for (uint i = 0; i < gl_SubgroupSize; i++)\n"
113 << " if (subgroupBallotBitExtract(mask, i))\n"
115 << " firstActive = i;\n"
119 << " tempResult |= (subgroupBroadcastFirst(data1[gl_SubgroupInvocationID]) == data1[firstActive]) ? 0x1 : 0;\n"
120 << " // make the firstActive invocation inactive now\n"
121 << " if (firstActive == gl_SubgroupInvocationID)\n"
123 << " for (uint i = 0; i < gl_SubgroupSize; i++)\n"
125 << " if (subgroupBallotBitExtract(mask, i))\n"
127 << " firstActive = i;\n"
131 << " tempResult |= (subgroupBroadcastFirst(data1[gl_SubgroupInvocationID]) == data1[firstActive]) ? 0x2 : 0;\n"
135 << " // the firstActive invocation didn't partake in the second result so set it to true\n"
136 << " tempResult |= 0x2;\n"
142 void initFrameBufferPrograms(SourceCollections& programCollection, CaseDefinition caseDef)
144 subgroups::setFragmentShaderFrameBuffer(programCollection);
146 if (SHADER_STAGE_VERTEX_BIT != caseDef.shaderStage)
147 subgroups::setVertexShaderFrameBuffer(programCollection);
149 std::string bdyStr = getBodySource(caseDef);
151 if (SHADER_STAGE_VERTEX_BIT == caseDef.shaderStage)
153 std::ostringstream vertex;
154 vertex << glu::getGLSLVersionDeclaration(glu::GLSL_VERSION_450)<<"\n"
155 << "#extension GL_KHR_shader_subgroup_ballot: enable\n"
156 << "layout(location = 0) in highp vec4 in_position;\n"
157 << "layout(location = 0) out float out_color;\n"
158 << "layout(binding = 0) uniform Buffer0\n"
160 << " " << subgroups::getFormatNameForGLSL(caseDef.format) << " data1[" << subgroups::maxSupportedSubgroupSize() << "];\n"
163 << "void main (void)\n"
166 << " out_color = float(tempResult);\n"
167 << " gl_Position = in_position;\n"
168 << " gl_PointSize = 1.0f;\n"
170 programCollection.add("vert") << glu::VertexSource(vertex.str());
172 else if (SHADER_STAGE_GEOMETRY_BIT == caseDef.shaderStage)
174 std::ostringstream geometry;
176 geometry << glu::getGLSLVersionDeclaration(glu::GLSL_VERSION_450)<<"\n"
177 << "#extension GL_KHR_shader_subgroup_ballot: enable\n"
178 << "layout(points) in;\n"
179 << "layout(points, max_vertices = 1) out;\n"
180 << "layout(location = 0) out float out_color;\n"
181 << "layout(binding = 0) uniform Buffer0\n"
183 << " " << subgroups::getFormatNameForGLSL(caseDef.format) << " data1[" <<subgroups::maxSupportedSubgroupSize() << "];\n"
186 << "void main (void)\n"
189 << " out_color = float(tempResult);\n"
190 << " gl_Position = gl_in[0].gl_Position;\n"
191 << " EmitVertex();\n"
192 << " EndPrimitive();\n"
195 programCollection.add("geometry") << glu::GeometrySource(geometry.str());
197 else if (SHADER_STAGE_TESS_CONTROL_BIT == caseDef.shaderStage)
199 std::ostringstream controlSource;
201 controlSource << glu::getGLSLVersionDeclaration(glu::GLSL_VERSION_450)<<"\n"
202 << "#extension GL_KHR_shader_subgroup_ballot: enable\n"
203 << "layout(vertices = 2) out;\n"
204 << "layout(location = 0) out float out_color[];\n"
205 << "layout(binding = 0) uniform Buffer0\n"
207 << " " << subgroups::getFormatNameForGLSL(caseDef.format) << " data1[" <<subgroups::maxSupportedSubgroupSize() << "];\n"
210 << "void main (void)\n"
212 << " if (gl_InvocationID == 0)\n"
214 << " gl_TessLevelOuter[0] = 1.0f;\n"
215 << " gl_TessLevelOuter[1] = 1.0f;\n"
218 << " out_color[gl_InvocationID ] = float(tempResult);\n"
219 << " gl_out[gl_InvocationID].gl_Position = gl_in[gl_InvocationID].gl_Position;\n"
222 programCollection.add("tesc") << glu::TessellationControlSource(controlSource.str());
223 subgroups::setTesEvalShaderFrameBuffer(programCollection);
225 else if (SHADER_STAGE_TESS_EVALUATION_BIT == caseDef.shaderStage)
227 std::ostringstream evaluationSource;
228 evaluationSource << glu::getGLSLVersionDeclaration(glu::GLSL_VERSION_450)<<"\n"
229 << "#extension GL_KHR_shader_subgroup_ballot: enable\n"
230 << "layout(isolines, equal_spacing, ccw ) in;\n"
231 << "layout(location = 0) out float out_color;\n"
232 << "layout(binding = 0) uniform Buffer0\n"
234 << " " << subgroups::getFormatNameForGLSL(caseDef.format) << " data1[" <<subgroups::maxSupportedSubgroupSize() << "];\n"
237 << "void main (void)\n"
240 << " out_color = float(tempResult);\n"
241 << " gl_Position = mix(gl_in[0].gl_Position, gl_in[1].gl_Position, gl_TessCoord.x);\n"
244 subgroups::setTesCtrlShaderFrameBuffer(programCollection);
245 programCollection.add("tese") << glu::TessellationEvaluationSource(evaluationSource.str());
249 DE_FATAL("Unsupported shader stage");
253 void initPrograms(SourceCollections& programCollection, CaseDefinition caseDef)
255 std::string bdyStr = getBodySource(caseDef);
257 if (SHADER_STAGE_COMPUTE_BIT == caseDef.shaderStage)
259 std::ostringstream src;
261 src << "#version 450\n"
262 << "#extension GL_KHR_shader_subgroup_ballot: enable\n"
263 << "layout (${LOCAL_SIZE_X}, ${LOCAL_SIZE_Y}, ${LOCAL_SIZE_Z}) in;\n"
264 << "layout(binding = 0, std430) buffer Buffer0\n"
266 << " uint result[];\n"
268 << "layout(binding = 1, std430) buffer Buffer1\n"
270 << " " << subgroups::getFormatNameForGLSL(caseDef.format) << " data1[];\n"
273 << "void main (void)\n"
275 << " uvec3 globalSize = gl_NumWorkGroups * gl_WorkGroupSize;\n"
276 << " highp uint offset = globalSize.x * ((globalSize.y * "
277 "gl_GlobalInvocationID.z) + gl_GlobalInvocationID.y) + "
278 "gl_GlobalInvocationID.x;\n"
280 << " result[offset] = tempResult;\n"
283 programCollection.add("comp") << glu::ComputeSource(src.str());
287 const string vertex =
289 "#extension GL_KHR_shader_subgroup_ballot: enable\n"
290 "layout(binding = 0, std430) buffer Buffer0\n"
294 "layout(set = 0, binding = 4, std430) readonly buffer Buffer4\n"
296 " " + subgroups::getFormatNameForGLSL(caseDef.format) + " data1[];\n"
302 " b0.result[gl_VertexID] = tempResult;\n"
303 " float pixelSize = 2.0f/1024.0f;\n"
304 " float pixelPosition = pixelSize/2.0f - 1.0f;\n"
305 " gl_Position = vec4(float(gl_VertexID) * pixelSize + pixelPosition, 0.0f, 0.0f, 1.0f);\n"
306 " gl_PointSize = 1.0f;\n"
311 "#extension GL_KHR_shader_subgroup_ballot: enable\n"
312 "layout(vertices=1) out;\n"
313 "layout(binding = 1, std430) buffer Buffer1\n"
317 "layout(binding = 4, std430) readonly buffer Buffer4\n"
319 " " + subgroups::getFormatNameForGLSL(caseDef.format) + " data1[];\n"
325 " b1.result[gl_PrimitiveID] = tempResult;\n"
326 " if (gl_InvocationID == 0)\n"
328 " gl_TessLevelOuter[0] = 1.0f;\n"
329 " gl_TessLevelOuter[1] = 1.0f;\n"
331 " gl_out[gl_InvocationID].gl_Position = gl_in[gl_InvocationID].gl_Position;\n"
336 "#extension GL_KHR_shader_subgroup_ballot: enable\n"
337 "layout(isolines) in;\n"
338 "layout(binding = 2, std430) buffer Buffer2\n"
342 "layout(binding = 4, std430) readonly buffer Buffer4\n"
344 " " + subgroups::getFormatNameForGLSL(caseDef.format) + " data1[];\n"
350 " b2.result[gl_PrimitiveID * 2 + uint(gl_TessCoord.x + 0.5)] = tempResult;\n"
351 " float pixelSize = 2.0f/1024.0f;\n"
352 " gl_Position = gl_in[0].gl_Position + gl_TessCoord.x * pixelSize / 2.0f;\n"
355 const string geometry =
357 "#extension GL_KHR_shader_subgroup_ballot: enable\n"
358 "layout(${TOPOLOGY}) in;\n"
359 "layout(points, max_vertices = 1) out;\n"
360 "layout(binding = 3, std430) buffer Buffer3\n"
364 "layout(set = 0, binding = 4, std430) readonly buffer Buffer4\n"
366 " " + subgroups::getFormatNameForGLSL(caseDef.format) + " data1[];\n"
372 " b3.result[gl_PrimitiveIDIn] = tempResult;\n"
373 " gl_Position = gl_in[0].gl_Position;\n"
378 const string fragment =
380 "#extension GL_KHR_shader_subgroup_ballot: enable\n"
381 "layout(location = 0) out uint result;\n"
382 "layout(binding = 4, std430) readonly buffer Buffer4\n"
384 " " + subgroups::getFormatNameForGLSL(caseDef.format) + " data1[];\n"
389 " result = tempResult;\n"
392 subgroups::addNoSubgroupShader(programCollection);
394 programCollection.add("vert") << glu::VertexSource(vertex);
395 programCollection.add("tesc") << glu::TessellationControlSource(tesc);
396 programCollection.add("tese") << glu::TessellationEvaluationSource(tese);
397 subgroups::addGeometryShadersFromTemplate(geometry, programCollection);
398 programCollection.add("fragment") << glu::FragmentSource(fragment);
402 void supportedCheck (Context& context, CaseDefinition caseDef)
404 if (!subgroups::isSubgroupSupported(context))
405 TCU_THROW(NotSupportedError, "Subgroup operations are not supported");
407 if (!subgroups::isSubgroupFeatureSupportedForDevice(context, SUBGROUP_FEATURE_BALLOT_BIT))
409 TCU_THROW(NotSupportedError, "Device does not support subgroup ballot operations");
412 if (subgroups::isDoubleFormat(caseDef.format) &&
413 !subgroups::isDoubleSupportedForDevice(context))
415 TCU_THROW(NotSupportedError, "Device does not support subgroup double operations");
419 tcu::TestStatus noSSBOtest (Context& context, const CaseDefinition caseDef)
421 if (!subgroups::areSubgroupOperationsSupportedForStage(
422 context, caseDef.shaderStage))
424 if (subgroups::areSubgroupOperationsRequiredForStage(caseDef.shaderStage))
426 return tcu::TestStatus::fail(
428 subgroups::getShaderStageName(caseDef.shaderStage) +
429 " is required to support subgroup operations!");
433 TCU_THROW(NotSupportedError, "Device does not support subgroup operations for this stage");
437 subgroups::SSBOData inputData[1];
438 inputData[0].format = caseDef.format;
439 inputData[0].numElements = subgroups::maxSupportedSubgroupSize();
440 inputData[0].initializeType = subgroups::SSBOData::InitializeNonZero;
442 if (SHADER_STAGE_VERTEX_BIT == caseDef.shaderStage)
443 return subgroups::makeVertexFrameBufferTest(context, FORMAT_R32_UINT, inputData, 1, checkVertexPipelineStages);
444 else if (SHADER_STAGE_GEOMETRY_BIT == caseDef.shaderStage)
445 return subgroups::makeGeometryFrameBufferTest(context, FORMAT_R32_UINT, inputData, 1, checkVertexPipelineStages);
446 else if (SHADER_STAGE_TESS_CONTROL_BIT == caseDef.shaderStage)
447 return subgroups::makeTessellationEvaluationFrameBufferTest(context, FORMAT_R32_UINT, inputData, 1, checkVertexPipelineStages, SHADER_STAGE_TESS_CONTROL_BIT);
448 else if (SHADER_STAGE_TESS_EVALUATION_BIT == caseDef.shaderStage)
449 return subgroups::makeTessellationEvaluationFrameBufferTest(context, FORMAT_R32_UINT, inputData, 1, checkVertexPipelineStages, SHADER_STAGE_TESS_EVALUATION_BIT);
451 TCU_THROW(InternalError, "Unhandled shader stage");
455 tcu::TestStatus test(Context& context, const CaseDefinition caseDef)
457 if (SHADER_STAGE_COMPUTE_BIT == caseDef.shaderStage)
459 if (!subgroups::areSubgroupOperationsSupportedForStage(context, caseDef.shaderStage))
461 if (subgroups::areSubgroupOperationsRequiredForStage(caseDef.shaderStage))
463 return tcu::TestStatus::fail(
465 subgroups::getShaderStageName(caseDef.shaderStage) +
466 " is required to support subgroup operations!");
470 TCU_THROW(NotSupportedError, "Device does not support subgroup operations for this stage");
473 subgroups::SSBOData inputData[1];
474 inputData[0].format = caseDef.format;
475 inputData[0].numElements = subgroups::maxSupportedSubgroupSize();
476 inputData[0].initializeType = subgroups::SSBOData::InitializeNonZero;
477 inputData[0].binding = 1u;
479 return subgroups::makeComputeTest(context, FORMAT_R32_UINT, inputData, 1, checkComputeStages);
483 int supportedStages = context.getDeqpContext().getContextInfo().getInt(GL_SUBGROUP_SUPPORTED_STAGES_KHR);
485 ShaderStageFlags stages = (ShaderStageFlags)(caseDef.shaderStage & supportedStages);
487 if ( SHADER_STAGE_FRAGMENT_BIT != stages && !subgroups::isVertexSSBOSupportedForDevice(context))
489 if ( (stages & SHADER_STAGE_FRAGMENT_BIT) == 0)
490 TCU_THROW(NotSupportedError, "Device does not support vertex stage SSBO writes");
492 stages = SHADER_STAGE_FRAGMENT_BIT;
495 if ((ShaderStageFlags)0u == stages)
496 TCU_THROW(NotSupportedError, "Subgroup operations are not supported for any graphic shader");
498 subgroups::SSBOData inputData;
499 inputData.format = caseDef.format;
500 inputData.numElements = subgroups::maxSupportedSubgroupSize();
501 inputData.initializeType = subgroups::SSBOData::InitializeNonZero;
502 inputData.binding = 4u;
503 inputData.stages = stages;
505 return subgroups::allStages(context, FORMAT_R32_UINT, &inputData, 1, checkVertexPipelineStages, stages);
510 deqp::TestCaseGroup* createSubgroupsBallotBroadcastTests(deqp::Context& testCtx)
512 de::MovePtr<deqp::TestCaseGroup> graphicGroup(new deqp::TestCaseGroup(
513 testCtx, "graphics", "Subgroup ballot broadcast category tests: graphics"));
514 de::MovePtr<deqp::TestCaseGroup> computeGroup(new deqp::TestCaseGroup(
515 testCtx, "compute", "Subgroup ballot broadcast category tests: compute"));
516 de::MovePtr<deqp::TestCaseGroup> framebufferGroup(new deqp::TestCaseGroup(
517 testCtx, "framebuffer", "Subgroup ballot broadcast category tests: framebuffer"));
519 const ShaderStageFlags stages[] =
521 SHADER_STAGE_VERTEX_BIT,
522 SHADER_STAGE_TESS_EVALUATION_BIT,
523 SHADER_STAGE_TESS_CONTROL_BIT,
524 SHADER_STAGE_GEOMETRY_BIT,
527 const Format formats[] =
529 FORMAT_R32_SINT, FORMAT_R32G32_SINT, FORMAT_R32G32B32_SINT,
530 FORMAT_R32G32B32A32_SINT, FORMAT_R32_UINT, FORMAT_R32G32_UINT,
531 FORMAT_R32G32B32_UINT, FORMAT_R32G32B32A32_UINT,
532 FORMAT_R32_SFLOAT, FORMAT_R32G32_SFLOAT,
533 FORMAT_R32G32B32_SFLOAT, FORMAT_R32G32B32A32_SFLOAT,
534 FORMAT_R64_SFLOAT, FORMAT_R64G64_SFLOAT,
535 FORMAT_R64G64B64_SFLOAT, FORMAT_R64G64B64A64_SFLOAT,
536 FORMAT_R32_BOOL, FORMAT_R32G32_BOOL,
537 FORMAT_R32G32B32_BOOL, FORMAT_R32G32B32A32_BOOL,
540 for (int formatIndex = 0; formatIndex < DE_LENGTH_OF_ARRAY(formats); ++formatIndex)
542 const Format format = formats[formatIndex];
544 for (int opTypeIndex = 0; opTypeIndex < OPTYPE_LAST; ++opTypeIndex)
546 const std::string op = de::toLower(getOpTypeName(opTypeIndex));
547 const std::string name = op + "_" + subgroups::getFormatNameForGLSL(format);
550 CaseDefinition caseDef = {opTypeIndex, SHADER_STAGE_COMPUTE_BIT, format};
551 SubgroupFactory<CaseDefinition>::addFunctionCaseWithPrograms(computeGroup.get(), name, "", supportedCheck, initPrograms, test, caseDef);
555 const CaseDefinition caseDef = {opTypeIndex, SHADER_STAGE_ALL_GRAPHICS, format};
556 SubgroupFactory<CaseDefinition>::addFunctionCaseWithPrograms(graphicGroup.get(), name, "", supportedCheck, initPrograms, test, caseDef);
559 for (int stageIndex = 0; stageIndex < DE_LENGTH_OF_ARRAY(stages); ++stageIndex)
561 const CaseDefinition caseDef = {opTypeIndex, stages[stageIndex], format};
562 SubgroupFactory<CaseDefinition>::addFunctionCaseWithPrograms(framebufferGroup.get(), name + getShaderStageName(caseDef.shaderStage), "",
563 supportedCheck, initFrameBufferPrograms, noSSBOtest, caseDef);
568 de::MovePtr<deqp::TestCaseGroup> group(new deqp::TestCaseGroup(
569 testCtx, "ballot_broadcast", "Subgroup ballot broadcast category tests"));
571 group->addChild(graphicGroup.release());
572 group->addChild(computeGroup.release());
573 group->addChild(framebufferGroup.release());
574 return group.release();