123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266 |
- //
- // Copyright (C) 2016-2018 Google, Inc.
- //
- // All rights reserved.
- //
- // Redistribution and use in source and binary forms, with or without
- // modification, are permitted provided that the following conditions
- // are met:
- //
- // Redistributions of source code must retain the above copyright
- // notice, this list of conditions and the following disclaimer.
- //
- // Redistributions in binary form must reproduce the above
- // copyright notice, this list of conditions and the following
- // disclaimer in the documentation and/or other materials provided
- // with the distribution.
- //
- // Neither the name of 3Dlabs Inc. Ltd. nor the names of its
- // contributors may be used to endorse or promote products derived
- // from this software without specific prior written permission.
- //
- // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
- // FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
- // COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
- // INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
- // BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- // LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- // LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
- // ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- // POSSIBILITY OF SUCH DAMAGE.
- //
- // Post-processing for SPIR-V IR, in internal form, not standard binary form.
- //
- #include <cassert>
- #include <cstdlib>
- #include <unordered_set>
- #include <algorithm>
- #include "SpvBuilder.h"
- #include "spirv.hpp"
- #include "GlslangToSpv.h"
- #include "SpvBuilder.h"
- namespace spv {
- #include "GLSL.std.450.h"
- #include "GLSL.ext.KHR.h"
- #include "GLSL.ext.EXT.h"
- #ifdef AMD_EXTENSIONS
- #include "GLSL.ext.AMD.h"
- #endif
- #ifdef NV_EXTENSIONS
- #include "GLSL.ext.NV.h"
- #endif
- }
- namespace spv {
- // Hook to visit each operand type and result type of an instruction.
- // Will be called multiple times for one instruction, once for each typed
- // operand and the result.
- void Builder::postProcessType(const Instruction& inst, Id typeId)
- {
- // Characterize the type being questioned
- Id basicTypeOp = getMostBasicTypeClass(typeId);
- int width = 0;
- if (basicTypeOp == OpTypeFloat || basicTypeOp == OpTypeInt)
- width = getScalarTypeWidth(typeId);
- // Do opcode-specific checks
- switch (inst.getOpCode()) {
- case OpLoad:
- case OpStore:
- if (basicTypeOp == OpTypeStruct) {
- if (containsType(typeId, OpTypeInt, 8))
- addCapability(CapabilityInt8);
- if (containsType(typeId, OpTypeInt, 16))
- addCapability(CapabilityInt16);
- if (containsType(typeId, OpTypeFloat, 16))
- addCapability(CapabilityFloat16);
- } else {
- StorageClass storageClass = getStorageClass(inst.getIdOperand(0));
- if (width == 8) {
- switch (storageClass) {
- case StorageClassUniform:
- case StorageClassStorageBuffer:
- case StorageClassPushConstant:
- break;
- default:
- addCapability(CapabilityInt8);
- break;
- }
- } else if (width == 16) {
- switch (storageClass) {
- case StorageClassUniform:
- case StorageClassStorageBuffer:
- case StorageClassPushConstant:
- case StorageClassInput:
- case StorageClassOutput:
- break;
- default:
- if (basicTypeOp == OpTypeInt)
- addCapability(CapabilityInt16);
- if (basicTypeOp == OpTypeFloat)
- addCapability(CapabilityFloat16);
- break;
- }
- }
- }
- break;
- case OpAccessChain:
- case OpPtrAccessChain:
- case OpCopyObject:
- case OpFConvert:
- case OpSConvert:
- case OpUConvert:
- break;
- case OpExtInst:
- switch (inst.getImmediateOperand(1)) {
- #if AMD_EXTENSIONS
- case GLSLstd450Frexp:
- case GLSLstd450FrexpStruct:
- if (getSpvVersion() < glslang::EShTargetSpv_1_3 && containsType(typeId, OpTypeInt, 16))
- addExtension(spv::E_SPV_AMD_gpu_shader_int16);
- break;
- case GLSLstd450InterpolateAtCentroid:
- case GLSLstd450InterpolateAtSample:
- case GLSLstd450InterpolateAtOffset:
- if (getSpvVersion() < glslang::EShTargetSpv_1_3 && containsType(typeId, OpTypeFloat, 16))
- addExtension(spv::E_SPV_AMD_gpu_shader_half_float);
- break;
- #endif
- default:
- break;
- }
- break;
- default:
- if (basicTypeOp == OpTypeFloat && width == 16)
- addCapability(CapabilityFloat16);
- if (basicTypeOp == OpTypeInt && width == 16)
- addCapability(CapabilityInt16);
- if (basicTypeOp == OpTypeInt && width == 8)
- addCapability(CapabilityInt8);
- break;
- }
- }
- // Called for each instruction that resides in a block.
- void Builder::postProcess(const Instruction& inst)
- {
- // Add capabilities based simply on the opcode.
- switch (inst.getOpCode()) {
- case OpExtInst:
- switch (inst.getImmediateOperand(1)) {
- case GLSLstd450InterpolateAtCentroid:
- case GLSLstd450InterpolateAtSample:
- case GLSLstd450InterpolateAtOffset:
- addCapability(CapabilityInterpolationFunction);
- break;
- default:
- break;
- }
- break;
- case OpDPdxFine:
- case OpDPdyFine:
- case OpFwidthFine:
- case OpDPdxCoarse:
- case OpDPdyCoarse:
- case OpFwidthCoarse:
- addCapability(CapabilityDerivativeControl);
- break;
- case OpImageQueryLod:
- case OpImageQuerySize:
- case OpImageQuerySizeLod:
- case OpImageQuerySamples:
- case OpImageQueryLevels:
- addCapability(CapabilityImageQuery);
- break;
- #ifdef NV_EXTENSIONS
- case OpGroupNonUniformPartitionNV:
- addExtension(E_SPV_NV_shader_subgroup_partitioned);
- addCapability(CapabilityGroupNonUniformPartitionedNV);
- break;
- #endif
- default:
- break;
- }
- // Checks based on type
- if (inst.getTypeId() != NoType)
- postProcessType(inst, inst.getTypeId());
- for (int op = 0; op < inst.getNumOperands(); ++op) {
- if (inst.isIdOperand(op)) {
- // In blocks, these are always result ids, but we are relying on
- // getTypeId() to return NoType for things like OpLabel.
- if (getTypeId(inst.getIdOperand(op)) != NoType)
- postProcessType(inst, getTypeId(inst.getIdOperand(op)));
- }
- }
- }
- // Called for each instruction in a reachable block.
- void Builder::postProcessReachable(const Instruction& inst)
- {
- // did have code here, but questionable to do so without deleting the instructions
- }
- // comment in header
- void Builder::postProcess()
- {
- std::unordered_set<const Block*> reachableBlocks;
- std::unordered_set<Id> unreachableDefinitions;
- // Collect IDs defined in unreachable blocks. For each function, label the
- // reachable blocks first. Then for each unreachable block, collect the
- // result IDs of the instructions in it.
- for (auto fi = module.getFunctions().cbegin(); fi != module.getFunctions().cend(); fi++) {
- Function* f = *fi;
- Block* entry = f->getEntryBlock();
- inReadableOrder(entry, [&reachableBlocks](const Block* b) { reachableBlocks.insert(b); });
- for (auto bi = f->getBlocks().cbegin(); bi != f->getBlocks().cend(); bi++) {
- Block* b = *bi;
- if (reachableBlocks.count(b) == 0) {
- for (auto ii = b->getInstructions().cbegin(); ii != b->getInstructions().cend(); ii++)
- unreachableDefinitions.insert(ii->get()->getResultId());
- }
- }
- }
- // Remove unneeded decorations, for unreachable instructions
- decorations.erase(std::remove_if(decorations.begin(), decorations.end(),
- [&unreachableDefinitions](std::unique_ptr<Instruction>& I) -> bool {
- Id decoration_id = I.get()->getIdOperand(0);
- return unreachableDefinitions.count(decoration_id) != 0;
- }),
- decorations.end());
- // Add per-instruction capabilities, extensions, etc.,
- // process all reachable instructions...
- for (auto bi = reachableBlocks.cbegin(); bi != reachableBlocks.cend(); ++bi) {
- const Block* block = *bi;
- const auto function = [this](const std::unique_ptr<Instruction>& inst) { postProcessReachable(*inst.get()); };
- std::for_each(block->getInstructions().begin(), block->getInstructions().end(), function);
- }
- // process all block-contained instructions
- for (auto fi = module.getFunctions().cbegin(); fi != module.getFunctions().cend(); fi++) {
- Function* f = *fi;
- for (auto bi = f->getBlocks().cbegin(); bi != f->getBlocks().cend(); bi++) {
- Block* b = *bi;
- for (auto ii = b->getInstructions().cbegin(); ii != b->getInstructions().cend(); ii++)
- postProcess(*ii->get());
- }
- }
- }
- }; // end spv namespace
|