SPIRV-Tools/source/val/validate_derivatives.cpp
2018-09-19 20:46:14 -04:00

99 lines
3.1 KiB
C++

// Copyright (c) 2017 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Validates correctness of derivative SPIR-V instructions.
#include "source/val/validate.h"
#include <string>
#include "source/diagnostic.h"
#include "source/opcode.h"
#include "source/val/instruction.h"
#include "source/val/validation_state.h"
namespace spvtools {
namespace val {
// Validates correctness of derivative instructions.
spv_result_t DerivativesPass(ValidationState_t& _, const Instruction* inst) {
const SpvOp opcode = inst->opcode();
const uint32_t result_type = inst->type_id();
switch (opcode) {
case SpvOpDPdx:
case SpvOpDPdy:
case SpvOpFwidth:
case SpvOpDPdxFine:
case SpvOpDPdyFine:
case SpvOpFwidthFine:
case SpvOpDPdxCoarse:
case SpvOpDPdyCoarse:
case SpvOpFwidthCoarse: {
if (!_.IsFloatScalarOrVectorType(result_type)) {
return _.diag(SPV_ERROR_INVALID_DATA, inst)
<< "Expected Result Type to be float scalar or vector type: "
<< spvOpcodeString(opcode);
}
const uint32_t p_type = _.GetOperandTypeId(inst, 2);
if (p_type != result_type) {
return _.diag(SPV_ERROR_INVALID_DATA, inst)
<< "Expected P type and Result Type to be the same: "
<< spvOpcodeString(opcode);
}
const spvtools::Extension compute_shader_derivatives_extension =
kSPV_NV_compute_shader_derivatives;
ExtensionSet exts(1, &compute_shader_derivatives_extension);
if (_.HasAnyOfExtensions(exts)) {
_.function(inst->function()->id())
->RegisterExecutionModelLimitation([opcode](SpvExecutionModel model,
std::string* message) {
if (model != SpvExecutionModelFragment &&
model != SpvExecutionModelGLCompute) {
if (message) {
*message =
std::string(
"Derivative instructions require Fragment execution "
"model: ") +
spvOpcodeString(opcode);
}
return false;
}
return true;
});
} else {
_.function(inst->function()->id())
->RegisterExecutionModelLimitation(
SpvExecutionModelFragment,
std::string(
"Derivative instructions require Fragment execution "
"model: ") +
spvOpcodeString(opcode));
}
break;
}
default:
break;
}
return SPV_SUCCESS;
}
} // namespace val
} // namespace spvtools