// Copyright 2019 Google LLC
//
// 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
//
//      https://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.

// Opcode table for the V0 binary format.
// Additions are fine but changing the behavior or order of any opcodes will
// break parsing of existing files.
//
// Opcodes have been selected on frequency of use, general applicability, and
// relative stability. Experimental ops should be implemented via the Foreign
// Function Interface (FFI) first before graduating into the core set. Ops that
// may only be present on certain targets should also be kept as imports via the
// FFI.
//
// Opcodes may be specified for particular types (int32_t), categories of types
// (all floating-point types), or implicit types (output matches input). Saving
// opcode space by sharing a single opcode for multiple types is preferred
// except where hot operations are performed (for example, comparison used in
// loop iterators).

#ifndef THIRD_PARTY_MLIR_EDGE_IREE_SCHEMAS_BYTECODE_INTERPRETER_BYTECODE_V0_H_
#define THIRD_PARTY_MLIR_EDGE_IREE_SCHEMAS_BYTECODE_INTERPRETER_BYTECODE_V0_H_

#include "third_party/mlir_edge/iree/schemas/bytecode/bytecode_v0.h"

namespace iree {

#define IREE_INTERPRETER_OPCODE_LIST(OPC, RESERVED_OPC)                       \
  OPC(0x00, kConstant, "constant", FLAG(kDefault), "cr", FF)                  \
                                                                              \
  OPC(0x01, kCall, "call", FLAG(kDefault), "fSR", FF)                         \
  OPC(0x02, kCallImport, "call_import", FLAG(kDefault), "FSR", FF)            \
  OPC(0x03, kCallIndirect, "call_indirect", FLAG(kDefault), "tsSR", FF)       \
  OPC(0x04, kReturn, "return", FLAG(kDefault), "S", FF)                       \
  OPC(0x05, kBranch, "br", FLAG(kDefault), "bT", FF)                          \
  OPC(0x06, kCondBranch, "cond_br", FLAG(kDefault), "sbTbT", FF)              \
  OPC(0x07, kCmpI, "cmp_i", FLAG(kDefault), "psso", FF)                       \
  OPC(0x08, kCmpF, "cmp_f", FLAG(kDefault), "Psso", FF)                       \
                                                                              \
  RSV(0x09, RESERVED_OPC)                                                     \
  RSV(0x0A, RESERVED_OPC)                                                     \
  RSV(0x0B, RESERVED_OPC)                                                     \
  RSV(0x0C, RESERVED_OPC)                                                     \
  RSV(0x0D, RESERVED_OPC)                                                     \
  RSV(0x0E, RESERVED_OPC)                                                     \
  RSV(0x0F, RESERVED_OPC)                                                     \
  RSV(0x10, RESERVED_OPC)                                                     \
  RSV(0x11, RESERVED_OPC)                                                     \
  RSV(0x12, RESERVED_OPC)                                                     \
  RSV(0x13, RESERVED_OPC)                                                     \
  RSV(0x14, RESERVED_OPC)                                                     \
  RSV(0x15, RESERVED_OPC)                                                     \
  RSV(0x16, RESERVED_OPC)                                                     \
  RSV(0x17, RESERVED_OPC)                                                     \
  RSV(0x18, RESERVED_OPC)                                                     \
  RSV(0x19, RESERVED_OPC)                                                     \
  RSV(0x1A, RESERVED_OPC)                                                     \
  RSV(0x1B, RESERVED_OPC)                                                     \
  RSV(0x1C, RESERVED_OPC)                                                     \
  RSV(0x1D, RESERVED_OPC)                                                     \
  RSV(0x1E, RESERVED_OPC)                                                     \
  RSV(0x1F, RESERVED_OPC)                                                     \
                                                                              \
  OPC(0x20, kAllocStatic, "alloc_static", FLAG(kDefault), "Icr", FF)          \
  OPC(0x21, kAllocStack, "alloc_stack", FLAG(kDefault), "itISr", FF)          \
  OPC(0x22, kAllocStackInit, "alloc_stack_init", FLAG(kDefault), "tIScr", FF) \
  OPC(0x23, kAllocHeap, "alloc_heap", FLAG(kDefault), "itISr", FF)            \
  OPC(0x24, kDiscard, "discard", FLAG(kDefault), "s", FF)                     \
                                                                              \
  RSV(0x25, RESERVED_OPC)                                                     \
  RSV(0x26, RESERVED_OPC)                                                     \
  RSV(0x27, RESERVED_OPC)                                                     \
  RSV(0x28, RESERVED_OPC)                                                     \
  RSV(0x29, RESERVED_OPC)                                                     \
  RSV(0x2A, RESERVED_OPC)                                                     \
  RSV(0x2B, RESERVED_OPC)                                                     \
  RSV(0x2C, RESERVED_OPC)                                                     \
  RSV(0x2D, RESERVED_OPC)                                                     \
  RSV(0x2E, RESERVED_OPC)                                                     \
  RSV(0x2F, RESERVED_OPC)                                                     \
                                                                              \
  OPC(0x30, kRank, "rank", FLAG(kDefault), "so", FF)                          \
  OPC(0x31, kDim, "dim", FLAG(kDefault), "iso", FF)                           \
  OPC(0x32, kShape, "shape", FLAG(kDefault), "so", FF)                        \
  OPC(0x33, kLength, "length", FLAG(kDefault), "so", FF)                      \
  OPC(0x34, kDynamicSlice, "dynamic_slice", FLAG(kDefault), "sssr", FF)       \
  OPC(0x35, kStaticSlice, "static_slice", FLAG(kDefault), "sIIr", FF)         \
  OPC(0x36, kDynamicCopy, "dynamic_copy", FLAG(kDefault), "ssoss", FF)        \
  OPC(0x37, kStaticCopy, "static_copy", FLAG(kDefault), "sIoII", FF)          \
  OPC(0x38, kClone, "clone", FLAG(kDefault), "sr", FF)                        \
  RSV(0x39, RESERVED_OPC)                                                     \
  OPC(0x3A, kSplit, "split", FLAG(kDefault), "isR", FF)                       \
  OPC(0x3B, kAssign, "assign", FLAG(kDefault), "sr", FF)                      \
  OPC(0x3C, kCondAssign, "cond_assign", FLAG(kDefault), "sssr", FF)           \
  OPC(0x3D, kReshape, "reshape", FLAG(kDefault), "ssr", FF)                   \
  OPC(0x3E, kSelect, "select", FLAG(kDefault), "ssso", FF)                    \
  OPC(0x3F, kTranspose, "transpose", FLAG(kDefault), "sso", FF)               \
  OPC(0x40, kBroadcast, "broadcast", FLAG(kDefault), "sso", FF)               \
  OPC(0x41, kTile, "tile", FLAG(kDefault), "sso", FF)                         \
  OPC(0x42, kReverse, "reverse", FLAG(kDefault), "sso", FF)                   \
  OPC(0x43, kPad, "pad", FLAG(kDefault), "ssssso", FF)                        \
                                                                              \
  RSV(0x44, RESERVED_OPC)                                                     \
  RSV(0x45, RESERVED_OPC)                                                     \
  RSV(0x46, RESERVED_OPC)                                                     \
  RSV(0x47, RESERVED_OPC)                                                     \
  RSV(0x48, RESERVED_OPC)                                                     \
  RSV(0x49, RESERVED_OPC)                                                     \
  RSV(0x4A, RESERVED_OPC)                                                     \
  RSV(0x4B, RESERVED_OPC)                                                     \
  RSV(0x4C, RESERVED_OPC)                                                     \
  RSV(0x4D, RESERVED_OPC)                                                     \
  RSV(0x4E, RESERVED_OPC)                                                     \
  RSV(0x4F, RESERVED_OPC)                                                     \
                                                                              \
  OPC(0x50, kNot, "not", FLAG(kDefault), "so", FF)                            \
  OPC(0x51, kAnd, "and", FLAG(kDefault), "sso", FF)                           \
  OPC(0x52, kOr, "or", FLAG(kDefault), "sso", FF)                             \
  OPC(0x53, kXor, "xor", FLAG(kDefault), "sso", FF)                           \
  OPC(0x54, kShiftLeft, "sll", FLAG(kDefault), "sso", FF)                     \
  OPC(0x55, kShiftRightLogical, "srl", FLAG(kDefault), "sso", FF)             \
  OPC(0x56, kShiftRightArithmetic, "sra", FLAG(kDefault), "sso", FF)          \
                                                                              \
  RSV(0x57, RESERVED_OPC)                                                     \
  RSV(0x58, RESERVED_OPC)                                                     \
  RSV(0x59, RESERVED_OPC)                                                     \
  RSV(0x5A, RESERVED_OPC)                                                     \
  RSV(0x5B, RESERVED_OPC)                                                     \
  RSV(0x5C, RESERVED_OPC)                                                     \
  RSV(0x5D, RESERVED_OPC)                                                     \
  RSV(0x5E, RESERVED_OPC)                                                     \
  RSV(0x5F, RESERVED_OPC)                                                     \
  RSV(0x60, RESERVED_OPC)                                                     \
  RSV(0x61, RESERVED_OPC)                                                     \
  RSV(0x62, RESERVED_OPC)                                                     \
  RSV(0x63, RESERVED_OPC)                                                     \
  RSV(0x64, RESERVED_OPC)                                                     \
  RSV(0x65, RESERVED_OPC)                                                     \
  RSV(0x66, RESERVED_OPC)                                                     \
  RSV(0x67, RESERVED_OPC)                                                     \
  RSV(0x68, RESERVED_OPC)                                                     \
  RSV(0x69, RESERVED_OPC)                                                     \
  RSV(0x6A, RESERVED_OPC)                                                     \
  RSV(0x6B, RESERVED_OPC)                                                     \
  RSV(0x6C, RESERVED_OPC)                                                     \
  RSV(0x6D, RESERVED_OPC)                                                     \
  RSV(0x6E, RESERVED_OPC)                                                     \
  RSV(0x6F, RESERVED_OPC)                                                     \
                                                                              \
  /* TODO(benvanik): remove ones we don't need/can emulate */                 \
  OPC(0x70, kAddI, "add_i", FLAG(kDefault), "sso", FF)                        \
  OPC(0x71, kAddF, "add_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x72, kSubI, "sub_i", FLAG(kDefault), "sso", FF)                        \
  OPC(0x73, kSubF, "sub_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x74, kAbsI, "abs_i", FLAG(kDefault), "so", FF)                         \
  OPC(0x75, kAbsF, "abs_f", FLAG(kDefault), "so", FF)                         \
  OPC(0x76, kMulI, "mul_i", FLAG(kDefault), "sso", FF)                        \
  OPC(0x77, kMulF, "mul_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x78, kDivIS, "div_i_s", FLAG(kDefault), "sso", FF)                     \
  OPC(0x79, kDivIU, "div_i_u", FLAG(kDefault), "sso", FF)                     \
  OPC(0x7A, kDivF, "div_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x7B, kMulAddI, "madd_i", FLAG(kDefault), "ssso", FF)                   \
  OPC(0x7C, kMulAddF, "madd_f", FLAG(kDefault), "ssso", FF)                   \
  OPC(0x7D, kCosF, "cos_f", FLAG(kDefault), "so", FF)                         \
  OPC(0x7E, kSinF, "sin_f", FLAG(kDefault), "so", FF)                         \
  OPC(0x7F, kTanhF, "tanh_f", FLAG(kDefault), "so", FF)                       \
  OPC(0x80, kAtan2F, "atan2_f", FLAG(kDefault), "sso", FF)                    \
  OPC(0x81, kExpF, "exp_f", FLAG(kDefault), "so", FF)                         \
  OPC(0x82, kLogF, "log_f", FLAG(kDefault), "so", FF)                         \
  OPC(0x83, kRsqrtF, "rsqrt_f", FLAG(kDefault), "so", FF)                     \
                                                                              \
  RSV(0x84, RESERVED_OPC)                                                     \
  RSV(0x85, RESERVED_OPC)                                                     \
  RSV(0x86, RESERVED_OPC)                                                     \
  RSV(0x87, RESERVED_OPC)                                                     \
  RSV(0x88, RESERVED_OPC)                                                     \
  RSV(0x89, RESERVED_OPC)                                                     \
  RSV(0x8A, RESERVED_OPC)                                                     \
  RSV(0x8B, RESERVED_OPC)                                                     \
  RSV(0x8C, RESERVED_OPC)                                                     \
  RSV(0x8D, RESERVED_OPC)                                                     \
  RSV(0x8E, RESERVED_OPC)                                                     \
  RSV(0x8F, RESERVED_OPC)                                                     \
                                                                              \
  OPC(0x90, kMinIS, "min_i_s", FLAG(kDefault), "sso", FF)                     \
  OPC(0x91, kMinIU, "min_i_u", FLAG(kDefault), "sso", FF)                     \
  OPC(0x92, kMinF, "min_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x93, kMaxIS, "max_i_s", FLAG(kDefault), "sso", FF)                     \
  OPC(0x94, kMaxIU, "max_i_u", FLAG(kDefault), "sso", FF)                     \
  OPC(0x95, kMaxF, "max_f", FLAG(kDefault), "sso", FF)                        \
  OPC(0x96, kClampIS, "clamp_i_s", FLAG(kDefault), "ssso", FF)                \
  OPC(0x97, kClampIU, "clamp_i_u", FLAG(kDefault), "ssso", FF)                \
  OPC(0x98, kClampF, "clamp_f", FLAG(kDefault), "ssso", FF)                   \
  OPC(0x99, kFloorF, "floor_f", FLAG(kDefault), "so", FF)                     \
  OPC(0x9A, kCeilF, "ceil_f", FLAG(kDefault), "so", FF)                       \
                                                                              \
  OPC(0x9B, kConvertSS, "convert_s_s", FLAG(kDefault), "tsto", FF)            \
  OPC(0x9C, kConvertUU, "convert_u_u", FLAG(kDefault), "tsto", FF)            \
  OPC(0x9D, kConvertSU, "convert_s_u", FLAG(kDefault), "tsto", FF)            \
  OPC(0x9E, kConvertUS, "convert_u_s", FLAG(kDefault), "tsto", FF)            \
                                                                              \
  RSV(0x9F, RESERVED_OPC)                                                     \
                                                                              \
  /* TODO(benvanik): reduction/sum/etc */                                     \
  /* TODO(benvanik): sort */                                                  \
                                                                              \
  OPC(0xA0, kMatMulI, "matmul_i", FLAG(kDefault), "sssso", FF)                \
  OPC(0xA1, kMatMulF, "matmul_f", FLAG(kDefault), "sso", FF)                  \
  /* TODO(benvanik): convolution */                                           \
                                                                              \
  OPC(0xA2, kReduceSumI, "reduce_sum_i", FLAG(kDefault), "ssio", FF)          \
  OPC(0xA3, kReduceSumF, "reduce_sum_f", FLAG(kDefault), "ssio", FF)          \
  OPC(0xA4, kReduceMinI, "reduce_min_i", FLAG(kDefault), "ssio", FF)          \
  OPC(0xA5, kReduceMinF, "reduce_min_f", FLAG(kDefault), "ssio", FF)          \
  OPC(0xA6, kReduceMaxI, "reduce_max_i", FLAG(kDefault), "ssio", FF)          \
  OPC(0xA7, kReduceMaxF, "reduce_max_f", FLAG(kDefault), "ssio", FF)          \
  RSV(0xA8, RESERVED_OPC)                                                     \
  RSV(0xA9, RESERVED_OPC)                                                     \
  RSV(0xAA, RESERVED_OPC)                                                     \
  RSV(0xAB, RESERVED_OPC)                                                     \
  RSV(0xAC, RESERVED_OPC)                                                     \
  RSV(0xAD, RESERVED_OPC)                                                     \
  RSV(0xAE, RESERVED_OPC)                                                     \
  RSV(0xAF, RESERVED_OPC)                                                     \
  RSV(0xB0, RESERVED_OPC)                                                     \
  RSV(0xB1, RESERVED_OPC)                                                     \
  RSV(0xB2, RESERVED_OPC)                                                     \
  RSV(0xB3, RESERVED_OPC)                                                     \
  RSV(0xB4, RESERVED_OPC)                                                     \
  RSV(0xB5, RESERVED_OPC)                                                     \
  RSV(0xB6, RESERVED_OPC)                                                     \
  RSV(0xB7, RESERVED_OPC)                                                     \
  RSV(0xB8, RESERVED_OPC)                                                     \
  RSV(0xB9, RESERVED_OPC)                                                     \
  RSV(0xBA, RESERVED_OPC)                                                     \
  RSV(0xBB, RESERVED_OPC)                                                     \
  RSV(0xBC, RESERVED_OPC)                                                     \
  RSV(0xBD, RESERVED_OPC)                                                     \
  RSV(0xBE, RESERVED_OPC)                                                     \
  RSV(0xBF, RESERVED_OPC)                                                     \
  RSV(0xC0, RESERVED_OPC)                                                     \
  RSV(0xC1, RESERVED_OPC)                                                     \
  RSV(0xC2, RESERVED_OPC)                                                     \
  RSV(0xC3, RESERVED_OPC)                                                     \
  RSV(0xC4, RESERVED_OPC)                                                     \
  RSV(0xC5, RESERVED_OPC)                                                     \
  RSV(0xC6, RESERVED_OPC)                                                     \
  RSV(0xC7, RESERVED_OPC)                                                     \
  RSV(0xC8, RESERVED_OPC)                                                     \
  RSV(0xC9, RESERVED_OPC)                                                     \
  RSV(0xCA, RESERVED_OPC)                                                     \
  RSV(0xCB, RESERVED_OPC)                                                     \
  RSV(0xCC, RESERVED_OPC)                                                     \
  RSV(0xCD, RESERVED_OPC)                                                     \
  RSV(0xCE, RESERVED_OPC)                                                     \
  RSV(0xCF, RESERVED_OPC)                                                     \
  RSV(0xD0, RESERVED_OPC)                                                     \
  RSV(0xD1, RESERVED_OPC)                                                     \
  RSV(0xD2, RESERVED_OPC)                                                     \
  RSV(0xD3, RESERVED_OPC)                                                     \
  RSV(0xD4, RESERVED_OPC)                                                     \
  RSV(0xD5, RESERVED_OPC)                                                     \
  RSV(0xD6, RESERVED_OPC)                                                     \
  RSV(0xD7, RESERVED_OPC)                                                     \
  RSV(0xD8, RESERVED_OPC)                                                     \
  RSV(0xD9, RESERVED_OPC)                                                     \
  RSV(0xDA, RESERVED_OPC)                                                     \
  RSV(0xDB, RESERVED_OPC)                                                     \
  RSV(0xDC, RESERVED_OPC)                                                     \
  RSV(0xDD, RESERVED_OPC)                                                     \
  RSV(0xDE, RESERVED_OPC)                                                     \
  RSV(0xDF, RESERVED_OPC)                                                     \
  RSV(0xE0, RESERVED_OPC)                                                     \
  RSV(0xE1, RESERVED_OPC)                                                     \
  RSV(0xE2, RESERVED_OPC)                                                     \
  RSV(0xE3, RESERVED_OPC)                                                     \
  RSV(0xE4, RESERVED_OPC)                                                     \
  RSV(0xE5, RESERVED_OPC)                                                     \
  RSV(0xE6, RESERVED_OPC)                                                     \
  RSV(0xE7, RESERVED_OPC)                                                     \
  RSV(0xE8, RESERVED_OPC)                                                     \
  RSV(0xE9, RESERVED_OPC)                                                     \
  RSV(0xEA, RESERVED_OPC)                                                     \
  RSV(0xEB, RESERVED_OPC)                                                     \
  RSV(0xEC, RESERVED_OPC)                                                     \
  RSV(0xED, RESERVED_OPC)                                                     \
  RSV(0xEE, RESERVED_OPC)                                                     \
  RSV(0xEF, RESERVED_OPC)                                                     \
  RSV(0xF0, RESERVED_OPC)                                                     \
  RSV(0xF1, RESERVED_OPC)                                                     \
  RSV(0xF2, RESERVED_OPC)                                                     \
  RSV(0xF3, RESERVED_OPC)                                                     \
  RSV(0xF4, RESERVED_OPC)                                                     \
  RSV(0xF5, RESERVED_OPC)                                                     \
  RSV(0xF6, RESERVED_OPC)                                                     \
  RSV(0xF7, RESERVED_OPC)                                                     \
  RSV(0xF8, RESERVED_OPC)                                                     \
  RSV(0xF9, RESERVED_OPC)                                                     \
  RSV(0xFA, RESERVED_OPC)                                                     \
  RSV(0xFB, RESERVED_OPC)                                                     \
  RSV(0xFC, RESERVED_OPC)                                                     \
                                                                              \
  OPC(0xFD, kTrace, "trace", FLAG(kDefault), "s", FF)                         \
  OPC(0xFE, kCondBreak, "cond_break", FLAG(kDefault), "s", FF)                \
  OPC(0xFF, kBreak, "break", FLAG(kDefault), "", FF)

#define DECLARE_ENUM(ordinal, enum_name, ...) enum_name = ordinal,
enum class InterpreterOpcode : uint8_t {
  IREE_INTERPRETER_OPCODE_LIST(DECLARE_ENUM, DECLARE_ENUM)
};
#undef DECLARE_ENUM

}  // namespace iree

#endif  // THIRD_PARTY_MLIR_EDGE_IREE_SCHEMAS_BYTECODE_INTERPRETER_BYTECODE_V0_H_