mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-25 13:00:04 +00:00
636 lines
27 KiB
C++
636 lines
27 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 conversion instructions.
|
|
|
|
#include "source/opcode.h"
|
|
#include "source/spirv_constant.h"
|
|
#include "source/spirv_target_env.h"
|
|
#include "source/val/instruction.h"
|
|
#include "source/val/validate.h"
|
|
#include "source/val/validation_state.h"
|
|
|
|
namespace spvtools {
|
|
namespace val {
|
|
|
|
// Validates correctness of conversion instructions.
|
|
spv_result_t ConversionPass(ValidationState_t& _, const Instruction* inst) {
|
|
const spv::Op opcode = inst->opcode();
|
|
const uint32_t result_type = inst->type_id();
|
|
|
|
switch (opcode) {
|
|
case spv::Op::OpConvertFToU: {
|
|
if (!_.IsUnsignedIntScalarType(result_type) &&
|
|
!_.IsUnsignedIntVectorType(result_type) &&
|
|
!_.IsUnsignedIntCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected unsigned int scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type || (!_.IsFloatScalarType(input_type) &&
|
|
!_.IsFloatVectorType(input_type) &&
|
|
!_.IsFloatCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be float scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpConvertFToS: {
|
|
if (!_.IsIntScalarType(result_type) && !_.IsIntVectorType(result_type) &&
|
|
!_.IsIntCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected int scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type || (!_.IsFloatScalarType(input_type) &&
|
|
!_.IsFloatVectorType(input_type) &&
|
|
!_.IsFloatCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be float scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpConvertSToF:
|
|
case spv::Op::OpConvertUToF: {
|
|
if (!_.IsFloatScalarType(result_type) &&
|
|
!_.IsFloatVectorType(result_type) &&
|
|
!_.IsFloatCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected float scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type ||
|
|
(!_.IsIntScalarType(input_type) && !_.IsIntVectorType(input_type) &&
|
|
!_.IsIntCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be int scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpUConvert: {
|
|
if (!_.IsUnsignedIntScalarType(result_type) &&
|
|
!_.IsUnsignedIntVectorType(result_type) &&
|
|
!_.IsUnsignedIntCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected unsigned int scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type ||
|
|
(!_.IsIntScalarType(input_type) && !_.IsIntVectorType(input_type) &&
|
|
!_.IsIntCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be int scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (_.GetBitWidth(result_type) == _.GetBitWidth(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have different bit width from Result "
|
|
"Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpSConvert: {
|
|
if (!_.IsIntScalarType(result_type) && !_.IsIntVectorType(result_type) &&
|
|
!_.IsIntCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected int scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type ||
|
|
(!_.IsIntScalarType(input_type) && !_.IsIntVectorType(input_type) &&
|
|
!_.IsIntCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be int scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (_.GetBitWidth(result_type) == _.GetBitWidth(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have different bit width from Result "
|
|
"Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpFConvert: {
|
|
if (!_.IsFloatScalarType(result_type) &&
|
|
!_.IsFloatVectorType(result_type) &&
|
|
!_.IsFloatCooperativeMatrixType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected float scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type || (!_.IsFloatScalarType(input_type) &&
|
|
!_.IsFloatVectorType(input_type) &&
|
|
!_.IsFloatCooperativeMatrixType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be float scalar or vector: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.IsCooperativeMatrixType(result_type) ||
|
|
_.IsCooperativeMatrixType(input_type)) {
|
|
spv_result_t ret =
|
|
_.CooperativeMatrixShapesMatch(inst, result_type, input_type, true);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
} else {
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (_.GetBitWidth(result_type) == _.GetBitWidth(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have different bit width from Result "
|
|
"Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpQuantizeToF16: {
|
|
if ((!_.IsFloatScalarType(result_type) &&
|
|
!_.IsFloatVectorType(result_type)) ||
|
|
_.GetBitWidth(result_type) != 32)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 32-bit float scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (input_type != result_type)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input type to be equal to Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpConvertPtrToU: {
|
|
if (!_.IsUnsignedIntScalarType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected unsigned int scalar type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!_.IsPointerType(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer: " << spvOpcodeString(opcode);
|
|
|
|
if (_.addressing_model() == spv::AddressingModel::Logical)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Logical addressing not supported: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.addressing_model() ==
|
|
spv::AddressingModel::PhysicalStorageBuffer64) {
|
|
spv::StorageClass input_storage_class;
|
|
uint32_t input_data_type = 0;
|
|
_.GetPointerTypeInfo(input_type, &input_data_type,
|
|
&input_storage_class);
|
|
if (input_storage_class != spv::StorageClass::PhysicalStorageBuffer)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Pointer storage class must be PhysicalStorageBuffer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (spvIsVulkanEnv(_.context()->target_env)) {
|
|
if (_.GetBitWidth(result_type) != 64) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< _.VkErrorID(4710)
|
|
<< "PhysicalStorageBuffer64 addressing mode requires the "
|
|
"result integer type to have a 64-bit width for Vulkan "
|
|
"environment.";
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpSatConvertSToU:
|
|
case spv::Op::OpSatConvertUToS: {
|
|
if (!_.IsIntScalarType(result_type) && !_.IsIntVectorType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected int scalar or vector type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type ||
|
|
(!_.IsIntScalarType(input_type) && !_.IsIntVectorType(input_type)))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected int scalar or vector as input: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.GetDimension(result_type) != _.GetDimension(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same dimension as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpConvertUToPtr: {
|
|
if (!_.IsPointerType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type || !_.IsIntScalarType(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected int scalar as input: " << spvOpcodeString(opcode);
|
|
|
|
if (_.addressing_model() == spv::AddressingModel::Logical)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Logical addressing not supported: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (_.addressing_model() ==
|
|
spv::AddressingModel::PhysicalStorageBuffer64) {
|
|
spv::StorageClass result_storage_class;
|
|
uint32_t result_data_type = 0;
|
|
_.GetPointerTypeInfo(result_type, &result_data_type,
|
|
&result_storage_class);
|
|
if (result_storage_class != spv::StorageClass::PhysicalStorageBuffer)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Pointer storage class must be PhysicalStorageBuffer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (spvIsVulkanEnv(_.context()->target_env)) {
|
|
if (_.GetBitWidth(input_type) != 64) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< _.VkErrorID(4710)
|
|
<< "PhysicalStorageBuffer64 addressing mode requires the "
|
|
"input integer to have a 64-bit width for Vulkan "
|
|
"environment.";
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpPtrCastToGeneric: {
|
|
spv::StorageClass result_storage_class;
|
|
uint32_t result_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(result_type, &result_data_type,
|
|
&result_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (result_storage_class != spv::StorageClass::Generic)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to have storage class Generic: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
spv::StorageClass input_storage_class;
|
|
uint32_t input_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(input_type, &input_data_type,
|
|
&input_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer: " << spvOpcodeString(opcode);
|
|
|
|
if (input_storage_class != spv::StorageClass::Workgroup &&
|
|
input_storage_class != spv::StorageClass::CrossWorkgroup &&
|
|
input_storage_class != spv::StorageClass::Function)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have storage class Workgroup, "
|
|
<< "CrossWorkgroup or Function: " << spvOpcodeString(opcode);
|
|
|
|
if (result_data_type != input_data_type)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input and Result Type to point to the same type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpGenericCastToPtr: {
|
|
spv::StorageClass result_storage_class;
|
|
uint32_t result_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(result_type, &result_data_type,
|
|
&result_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (result_storage_class != spv::StorageClass::Workgroup &&
|
|
result_storage_class != spv::StorageClass::CrossWorkgroup &&
|
|
result_storage_class != spv::StorageClass::Function)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to have storage class Workgroup, "
|
|
<< "CrossWorkgroup or Function: " << spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
spv::StorageClass input_storage_class;
|
|
uint32_t input_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(input_type, &input_data_type,
|
|
&input_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer: " << spvOpcodeString(opcode);
|
|
|
|
if (input_storage_class != spv::StorageClass::Generic)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have storage class Generic: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (result_data_type != input_data_type)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input and Result Type to point to the same type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpGenericCastToPtrExplicit: {
|
|
spv::StorageClass result_storage_class;
|
|
uint32_t result_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(result_type, &result_data_type,
|
|
&result_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const auto target_storage_class =
|
|
inst->GetOperandAs<spv::StorageClass>(3);
|
|
if (result_storage_class != target_storage_class)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be of target storage class: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
spv::StorageClass input_storage_class;
|
|
uint32_t input_data_type = 0;
|
|
if (!_.GetPointerTypeInfo(input_type, &input_data_type,
|
|
&input_storage_class))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer: " << spvOpcodeString(opcode);
|
|
|
|
if (input_storage_class != spv::StorageClass::Generic)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have storage class Generic: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (result_data_type != input_data_type)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input and Result Type to point to the same type: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (target_storage_class != spv::StorageClass::Workgroup &&
|
|
target_storage_class != spv::StorageClass::CrossWorkgroup &&
|
|
target_storage_class != spv::StorageClass::Function)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected target storage class to be Workgroup, "
|
|
<< "CrossWorkgroup or Function: " << spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpBitcast: {
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have a type: " << spvOpcodeString(opcode);
|
|
|
|
const bool result_is_pointer = _.IsPointerType(result_type);
|
|
const bool result_is_int_scalar = _.IsIntScalarType(result_type);
|
|
const bool input_is_pointer = _.IsPointerType(input_type);
|
|
const bool input_is_int_scalar = _.IsIntScalarType(input_type);
|
|
|
|
const bool result_is_coopmat = _.IsCooperativeMatrixType(result_type);
|
|
const bool input_is_coopmat = _.IsCooperativeMatrixType(input_type);
|
|
|
|
if (!result_is_pointer && !result_is_int_scalar && !result_is_coopmat &&
|
|
!_.IsIntVectorType(result_type) &&
|
|
!_.IsFloatScalarType(result_type) &&
|
|
!_.IsFloatVectorType(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a pointer or int or float vector "
|
|
<< "or scalar type: " << spvOpcodeString(opcode);
|
|
|
|
if (!input_is_pointer && !input_is_int_scalar && !input_is_coopmat &&
|
|
!_.IsIntVectorType(input_type) && !_.IsFloatScalarType(input_type) &&
|
|
!_.IsFloatVectorType(input_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer or int or float vector "
|
|
<< "or scalar: " << spvOpcodeString(opcode);
|
|
|
|
if (result_is_coopmat != input_is_coopmat)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Cooperative matrix can only be cast to another cooperative "
|
|
<< "matrix: " << spvOpcodeString(opcode);
|
|
|
|
if (result_is_coopmat) {
|
|
spv_result_t ret = _.CooperativeMatrixShapesMatch(inst, result_type,
|
|
input_type, false);
|
|
if (ret != SPV_SUCCESS) return ret;
|
|
}
|
|
|
|
if (_.version() >= SPV_SPIRV_VERSION_WORD(1, 5) ||
|
|
_.HasExtension(kSPV_KHR_physical_storage_buffer)) {
|
|
const bool result_is_int_vector = _.IsIntVectorType(result_type);
|
|
const bool result_has_int32 =
|
|
_.ContainsSizedIntOrFloatType(result_type, spv::Op::OpTypeInt, 32);
|
|
const bool input_is_int_vector = _.IsIntVectorType(input_type);
|
|
const bool input_has_int32 =
|
|
_.ContainsSizedIntOrFloatType(input_type, spv::Op::OpTypeInt, 32);
|
|
if (result_is_pointer && !input_is_pointer && !input_is_int_scalar &&
|
|
!(input_is_int_vector && input_has_int32))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer, int scalar or 32-bit int "
|
|
"vector if Result Type is pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (input_is_pointer && !result_is_pointer && !result_is_int_scalar &&
|
|
!(result_is_int_vector && result_has_int32))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Pointer can only be converted to another pointer, int "
|
|
"scalar or 32-bit int vector: "
|
|
<< spvOpcodeString(opcode);
|
|
} else {
|
|
if (result_is_pointer && !input_is_pointer && !input_is_int_scalar)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to be a pointer or int scalar if Result "
|
|
"Type is pointer: "
|
|
<< spvOpcodeString(opcode);
|
|
|
|
if (input_is_pointer && !result_is_pointer && !result_is_int_scalar)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Pointer can only be converted to another pointer or int "
|
|
"scalar: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (!result_is_pointer && !input_is_pointer) {
|
|
const uint32_t result_size =
|
|
_.GetBitWidth(result_type) * _.GetDimension(result_type);
|
|
const uint32_t input_size =
|
|
_.GetBitWidth(input_type) * _.GetDimension(input_type);
|
|
if (result_size != input_size)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected input to have the same total bit width as "
|
|
<< "Result Type: " << spvOpcodeString(opcode);
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpConvertUToAccelerationStructureKHR: {
|
|
if (!_.IsAccelerationStructureType(result_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Result Type to be a Acceleration Structure: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!input_type || !_.IsUnsigned64BitHandle(input_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 64-bit uint scalar or 2-component 32-bit uint "
|
|
"vector as input: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpCooperativeMatrixConvertNV:
|
|
case spv::Op::OpCooperativeMatrixTransposeNV: {
|
|
if (!_.IsCooperativeMatrixType(result_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected cooperative matrix Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
const uint32_t input_type = _.GetOperandTypeId(inst, 2);
|
|
if (!_.IsCooperativeMatrixType(input_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected cooperative matrix type for Matrix input: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
bool swap_row_col = (opcode == spv::Op::OpCooperativeMatrixTransposeNV);
|
|
if (auto error = _.CooperativeMatrixShapesMatch(
|
|
inst, result_type, input_type, true, swap_row_col))
|
|
return error;
|
|
|
|
if (opcode == spv::Op::OpCooperativeMatrixConvertNV) {
|
|
if (_.FindDef(result_type)->GetOperandAs<uint32_t>(1) !=
|
|
_.FindDef(input_type)->GetOperandAs<uint32_t>(1)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Result Type and Matrix component types mismatch: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
}
|
|
|
|
if (opcode == spv::Op::OpCooperativeMatrixTransposeNV) {
|
|
if (!_.IsCooperativeMatrixBType(result_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Result Type must have UseB: " << spvOpcodeString(opcode);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (_.HasCapability(spv::Capability::Shader)) {
|
|
switch (inst->opcode()) {
|
|
case spv::Op::OpConvertFToU:
|
|
case spv::Op::OpConvertFToS:
|
|
case spv::Op::OpConvertSToF:
|
|
case spv::Op::OpConvertUToF:
|
|
case spv::Op::OpBitcast:
|
|
if (_.ContainsLimitedUseIntOrFloatType(inst->type_id()) ||
|
|
_.ContainsLimitedUseIntOrFloatType(_.GetOperandTypeId(inst, 2u))) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "8- or 16-bit types can only be used with width-only "
|
|
"conversions";
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
return SPV_SUCCESS;
|
|
}
|
|
|
|
} // namespace val
|
|
} // namespace spvtools
|