Function ValidateDot

source/val/validate_arithmetics.cpp:198–241

}

spv_result_t ValidateDot(ValidationState_t& _, const Instruction* inst) {
  const spv::Op opcode = inst->opcode();
  const uint32_t result_type = inst->type_id();
  if (!_.IsFloatScalarType(result_type))
    return _.diag(SPV_ERROR_INVALID_DATA, inst)
           << "Expected float scalar type as Result Type: "
           << spvOpcodeString(opcode);

  if (_.IsBfloat16ScalarType(result_type)) {
    if (!_.HasCapability(spv::Capability::BFloat16DotProductKHR)) {
      return _.diag(SPV_ERROR_INVALID_DATA, inst)
             << "OpDot Result Type <id> " << _.getIdName(result_type)
             << "requires BFloat16DotProductKHR be declared.";
    }
  }

  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 components: "
             << spvOpcodeString(opcode);
    }
  }
  return SPV_SUCCESS;
}

spv_result_t ValidateVectorTimesScalar(ValidationState_t& _,
                                       const Instruction* inst) {