123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551 |
- // 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.
- // Performs validation of arithmetic instructions.
- #include "source/val/validate.h"
- #include <vector>
- #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 arithmetic instructions.
- spv_result_t ArithmeticsPass(ValidationState_t& _, const Instruction* inst) {
- const SpvOp opcode = inst->opcode();
- const uint32_t result_type = inst->type_id();
- switch (opcode) {
- case SpvOpFAdd:
- case SpvOpFSub:
- case SpvOpFMul:
- case SpvOpFDiv:
- case SpvOpFRem:
- case SpvOpFMod:
- case SpvOpFNegate: {
- bool supportsCoopMat =
- (opcode != SpvOpFMul && opcode != SpvOpFRem && opcode != SpvOpFMod);
- if (!_.IsFloatScalarType(result_type) &&
- !_.IsFloatVectorType(result_type) &&
- !(supportsCoopMat && _.IsFloatCooperativeMatrixType(result_type)))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected floating scalar or vector type as Result Type: "
- << spvOpcodeString(opcode);
- for (size_t operand_index = 2; operand_index < inst->operands().size();
- ++operand_index) {
- if (_.GetOperandTypeId(inst, operand_index) != result_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected arithmetic operands to be of Result Type: "
- << spvOpcodeString(opcode) << " operand index "
- << operand_index;
- }
- break;
- }
- case SpvOpUDiv:
- case SpvOpUMod: {
- bool supportsCoopMat = (opcode == SpvOpUDiv);
- if (!_.IsUnsignedIntScalarType(result_type) &&
- !_.IsUnsignedIntVectorType(result_type) &&
- !(supportsCoopMat &&
- _.IsUnsignedIntCooperativeMatrixType(result_type)))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected unsigned int scalar or vector type as Result Type: "
- << spvOpcodeString(opcode);
- for (size_t operand_index = 2; operand_index < inst->operands().size();
- ++operand_index) {
- if (_.GetOperandTypeId(inst, operand_index) != result_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected arithmetic operands to be of Result Type: "
- << spvOpcodeString(opcode) << " operand index "
- << operand_index;
- }
- break;
- }
- case SpvOpISub:
- case SpvOpIAdd:
- case SpvOpIMul:
- case SpvOpSDiv:
- case SpvOpSMod:
- case SpvOpSRem:
- case SpvOpSNegate: {
- bool supportsCoopMat =
- (opcode != SpvOpIMul && opcode != SpvOpSRem && opcode != SpvOpSMod);
- if (!_.IsIntScalarType(result_type) && !_.IsIntVectorType(result_type) &&
- !(supportsCoopMat && _.IsIntCooperativeMatrixType(result_type)))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected int scalar or vector type as Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t dimension = _.GetDimension(result_type);
- const uint32_t bit_width = _.GetBitWidth(result_type);
- for (size_t operand_index = 2; operand_index < inst->operands().size();
- ++operand_index) {
- const uint32_t type_id = _.GetOperandTypeId(inst, operand_index);
- if (!type_id ||
- (!_.IsIntScalarType(type_id) && !_.IsIntVectorType(type_id) &&
- !(supportsCoopMat && _.IsIntCooperativeMatrixType(result_type))))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected int scalar or vector type as operand: "
- << spvOpcodeString(opcode) << " operand index "
- << operand_index;
- if (_.GetDimension(type_id) != dimension)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected arithmetic operands to have the same dimension "
- << "as Result Type: " << spvOpcodeString(opcode)
- << " operand index " << operand_index;
- if (_.GetBitWidth(type_id) != bit_width)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected arithmetic operands to have the same bit width "
- << "as Result Type: " << spvOpcodeString(opcode)
- << " operand index " << operand_index;
- }
- break;
- }
- case SpvOpDot: {
- if (!_.IsFloatScalarType(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float scalar type as Result Type: "
- << spvOpcodeString(opcode);
- uint32_t first_vector_num_components = 0;
- for (size_t operand_index = 2; operand_index < inst->operands().size();
- ++operand_index) {
- const uint32_t type_id = _.GetOperandTypeId(inst, operand_index);
- if (!type_id || !_.IsFloatVectorType(type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector as operand: "
- << spvOpcodeString(opcode) << " operand index "
- << operand_index;
- const uint32_t component_type = _.GetComponentType(type_id);
- if (component_type != result_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component type to be equal to Result Type: "
- << spvOpcodeString(opcode) << " operand index "
- << operand_index;
- const uint32_t num_components = _.GetDimension(type_id);
- if (operand_index == 2) {
- first_vector_num_components = num_components;
- } else if (num_components != first_vector_num_components) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected operands to have the same number of componenets: "
- << spvOpcodeString(opcode);
- }
- }
- break;
- }
- case SpvOpVectorTimesScalar: {
- if (!_.IsFloatVectorType(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t vector_type_id = _.GetOperandTypeId(inst, 2);
- if (result_type != vector_type_id)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected vector operand type to be equal to Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t component_type = _.GetComponentType(vector_type_id);
- const uint32_t scalar_type_id = _.GetOperandTypeId(inst, 3);
- if (component_type != scalar_type_id)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected scalar operand type to be equal to the component "
- << "type of the vector operand: " << spvOpcodeString(opcode);
- break;
- }
- case SpvOpMatrixTimesScalar: {
- if (!_.IsFloatMatrixType(result_type) &&
- !_.IsCooperativeMatrixType(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t matrix_type_id = _.GetOperandTypeId(inst, 2);
- if (result_type != matrix_type_id)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected matrix operand type to be equal to Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t component_type = _.GetComponentType(matrix_type_id);
- const uint32_t scalar_type_id = _.GetOperandTypeId(inst, 3);
- if (component_type != scalar_type_id)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected scalar operand type to be equal to the component "
- << "type of the matrix operand: " << spvOpcodeString(opcode);
- break;
- }
- case SpvOpVectorTimesMatrix: {
- const uint32_t vector_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t matrix_type_id = _.GetOperandTypeId(inst, 3);
- if (!_.IsFloatVectorType(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as Result Type: "
- << spvOpcodeString(opcode);
- const uint32_t res_component_type = _.GetComponentType(result_type);
- if (!vector_type_id || !_.IsFloatVectorType(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as left operand: "
- << spvOpcodeString(opcode);
- if (res_component_type != _.GetComponentType(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component types of Result Type and vector to be "
- << "equal: " << spvOpcodeString(opcode);
- uint32_t matrix_num_rows = 0;
- uint32_t matrix_num_cols = 0;
- uint32_t matrix_col_type = 0;
- uint32_t matrix_component_type = 0;
- if (!_.GetMatrixTypeInfo(matrix_type_id, &matrix_num_rows,
- &matrix_num_cols, &matrix_col_type,
- &matrix_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as right operand: "
- << spvOpcodeString(opcode);
- if (res_component_type != matrix_component_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component types of Result Type and matrix to be "
- << "equal: " << spvOpcodeString(opcode);
- if (matrix_num_cols != _.GetDimension(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of columns of the matrix to be equal to "
- << "Result Type vector size: " << spvOpcodeString(opcode);
- if (matrix_num_rows != _.GetDimension(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of rows of the matrix to be equal to the "
- << "vector operand size: " << spvOpcodeString(opcode);
- break;
- }
- case SpvOpMatrixTimesVector: {
- const uint32_t matrix_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t vector_type_id = _.GetOperandTypeId(inst, 3);
- if (!_.IsFloatVectorType(result_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as Result Type: "
- << spvOpcodeString(opcode);
- uint32_t matrix_num_rows = 0;
- uint32_t matrix_num_cols = 0;
- uint32_t matrix_col_type = 0;
- uint32_t matrix_component_type = 0;
- if (!_.GetMatrixTypeInfo(matrix_type_id, &matrix_num_rows,
- &matrix_num_cols, &matrix_col_type,
- &matrix_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as left operand: "
- << spvOpcodeString(opcode);
- if (result_type != matrix_col_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected column type of the matrix to be equal to Result "
- "Type: "
- << spvOpcodeString(opcode);
- if (!vector_type_id || !_.IsFloatVectorType(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as right operand: "
- << spvOpcodeString(opcode);
- if (matrix_component_type != _.GetComponentType(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component types of the operands to be equal: "
- << spvOpcodeString(opcode);
- if (matrix_num_cols != _.GetDimension(vector_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of columns of the matrix to be equal to the "
- << "vector size: " << spvOpcodeString(opcode);
- break;
- }
- case SpvOpMatrixTimesMatrix: {
- const uint32_t left_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t right_type_id = _.GetOperandTypeId(inst, 3);
- uint32_t res_num_rows = 0;
- uint32_t res_num_cols = 0;
- uint32_t res_col_type = 0;
- uint32_t res_component_type = 0;
- if (!_.GetMatrixTypeInfo(result_type, &res_num_rows, &res_num_cols,
- &res_col_type, &res_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as Result Type: "
- << spvOpcodeString(opcode);
- uint32_t left_num_rows = 0;
- uint32_t left_num_cols = 0;
- uint32_t left_col_type = 0;
- uint32_t left_component_type = 0;
- if (!_.GetMatrixTypeInfo(left_type_id, &left_num_rows, &left_num_cols,
- &left_col_type, &left_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as left operand: "
- << spvOpcodeString(opcode);
- uint32_t right_num_rows = 0;
- uint32_t right_num_cols = 0;
- uint32_t right_col_type = 0;
- uint32_t right_component_type = 0;
- if (!_.GetMatrixTypeInfo(right_type_id, &right_num_rows, &right_num_cols,
- &right_col_type, &right_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as right operand: "
- << spvOpcodeString(opcode);
- if (!_.IsFloatScalarType(res_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as Result Type: "
- << spvOpcodeString(opcode);
- if (res_col_type != left_col_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected column types of Result Type and left matrix to be "
- << "equal: " << spvOpcodeString(opcode);
- if (res_component_type != right_component_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component types of Result Type and right matrix to "
- "be "
- << "equal: " << spvOpcodeString(opcode);
- if (res_num_cols != right_num_cols)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of columns of Result Type and right matrix "
- "to "
- << "be equal: " << spvOpcodeString(opcode);
- if (left_num_cols != right_num_rows)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of columns of left matrix and number of "
- "rows "
- << "of right matrix to be equal: " << spvOpcodeString(opcode);
- assert(left_num_rows == res_num_rows);
- break;
- }
- case SpvOpOuterProduct: {
- const uint32_t left_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t right_type_id = _.GetOperandTypeId(inst, 3);
- uint32_t res_num_rows = 0;
- uint32_t res_num_cols = 0;
- uint32_t res_col_type = 0;
- uint32_t res_component_type = 0;
- if (!_.GetMatrixTypeInfo(result_type, &res_num_rows, &res_num_cols,
- &res_col_type, &res_component_type))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float matrix type as Result Type: "
- << spvOpcodeString(opcode);
- if (left_type_id != res_col_type)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected column type of Result Type to be equal to the type "
- << "of the left operand: " << spvOpcodeString(opcode);
- if (!right_type_id || !_.IsFloatVectorType(right_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected float vector type as right operand: "
- << spvOpcodeString(opcode);
- if (res_component_type != _.GetComponentType(right_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected component types of the operands to be equal: "
- << spvOpcodeString(opcode);
- if (res_num_cols != _.GetDimension(right_type_id))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected number of columns of the matrix to be equal to the "
- << "vector size of the right operand: "
- << spvOpcodeString(opcode);
- break;
- }
- case SpvOpIAddCarry:
- case SpvOpISubBorrow:
- case SpvOpUMulExtended:
- case SpvOpSMulExtended: {
- std::vector<uint32_t> result_types;
- if (!_.GetStructMemberTypes(result_type, &result_types))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected a struct as Result Type: "
- << spvOpcodeString(opcode);
- if (result_types.size() != 2)
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected Result Type struct to have two members: "
- << spvOpcodeString(opcode);
- if (opcode == SpvOpSMulExtended) {
- if (!_.IsIntScalarType(result_types[0]) &&
- !_.IsIntVectorType(result_types[0]))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected Result Type struct member types to be integer "
- "scalar "
- << "or vector: " << spvOpcodeString(opcode);
- } else {
- if (!_.IsUnsignedIntScalarType(result_types[0]) &&
- !_.IsUnsignedIntVectorType(result_types[0]))
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected Result Type struct member types to be unsigned "
- << "integer scalar or vector: " << spvOpcodeString(opcode);
- }
- if (result_types[0] != result_types[1])
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected Result Type struct member types to be identical: "
- << spvOpcodeString(opcode);
- const uint32_t left_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t right_type_id = _.GetOperandTypeId(inst, 3);
- if (left_type_id != result_types[0] || right_type_id != result_types[0])
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected both operands to be of Result Type member type: "
- << spvOpcodeString(opcode);
- break;
- }
- case SpvOpCooperativeMatrixMulAddNV: {
- const uint32_t D_type_id = _.GetOperandTypeId(inst, 1);
- const uint32_t A_type_id = _.GetOperandTypeId(inst, 2);
- const uint32_t B_type_id = _.GetOperandTypeId(inst, 3);
- const uint32_t C_type_id = _.GetOperandTypeId(inst, 4);
- if (!_.IsCooperativeMatrixType(A_type_id)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected cooperative matrix type as A Type: "
- << spvOpcodeString(opcode);
- }
- if (!_.IsCooperativeMatrixType(B_type_id)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected cooperative matrix type as B Type: "
- << spvOpcodeString(opcode);
- }
- if (!_.IsCooperativeMatrixType(C_type_id)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected cooperative matrix type as C Type: "
- << spvOpcodeString(opcode);
- }
- if (!_.IsCooperativeMatrixType(D_type_id)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Expected cooperative matrix type as Result Type: "
- << spvOpcodeString(opcode);
- }
- const auto A = _.FindDef(A_type_id);
- const auto B = _.FindDef(B_type_id);
- const auto C = _.FindDef(C_type_id);
- const auto D = _.FindDef(D_type_id);
- std::tuple<bool, bool, uint32_t> A_scope, B_scope, C_scope, D_scope,
- A_rows, B_rows, C_rows, D_rows, A_cols, B_cols, C_cols, D_cols;
- A_scope = _.EvalInt32IfConst(A->GetOperandAs<uint32_t>(2));
- B_scope = _.EvalInt32IfConst(B->GetOperandAs<uint32_t>(2));
- C_scope = _.EvalInt32IfConst(C->GetOperandAs<uint32_t>(2));
- D_scope = _.EvalInt32IfConst(D->GetOperandAs<uint32_t>(2));
- A_rows = _.EvalInt32IfConst(A->GetOperandAs<uint32_t>(3));
- B_rows = _.EvalInt32IfConst(B->GetOperandAs<uint32_t>(3));
- C_rows = _.EvalInt32IfConst(C->GetOperandAs<uint32_t>(3));
- D_rows = _.EvalInt32IfConst(D->GetOperandAs<uint32_t>(3));
- A_cols = _.EvalInt32IfConst(A->GetOperandAs<uint32_t>(4));
- B_cols = _.EvalInt32IfConst(B->GetOperandAs<uint32_t>(4));
- C_cols = _.EvalInt32IfConst(C->GetOperandAs<uint32_t>(4));
- D_cols = _.EvalInt32IfConst(D->GetOperandAs<uint32_t>(4));
- const auto notEqual = [](std::tuple<bool, bool, uint32_t> X,
- std::tuple<bool, bool, uint32_t> Y) {
- return (std::get<1>(X) && std::get<1>(Y) &&
- std::get<2>(X) != std::get<2>(Y));
- };
- if (notEqual(A_scope, B_scope) || notEqual(A_scope, C_scope) ||
- notEqual(A_scope, D_scope) || notEqual(B_scope, C_scope) ||
- notEqual(B_scope, D_scope) || notEqual(C_scope, D_scope)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Cooperative matrix scopes must match: "
- << spvOpcodeString(opcode);
- }
- if (notEqual(A_rows, C_rows) || notEqual(A_rows, D_rows) ||
- notEqual(C_rows, D_rows)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Cooperative matrix 'M' mismatch: "
- << spvOpcodeString(opcode);
- }
- if (notEqual(B_cols, C_cols) || notEqual(B_cols, D_cols) ||
- notEqual(C_cols, D_cols)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Cooperative matrix 'N' mismatch: "
- << spvOpcodeString(opcode);
- }
- if (notEqual(A_cols, B_rows)) {
- return _.diag(SPV_ERROR_INVALID_DATA, inst)
- << "Cooperative matrix 'K' mismatch: "
- << spvOpcodeString(opcode);
- }
- break;
- }
- default:
- break;
- }
- return SPV_SUCCESS;
- }
- } // namespace val
- } // namespace spvtools
|