mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-01-26 12:10:22 +01:00
agx: Add opcode descriptions as Python
Pattern lifted from NIR. Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io> Acked-by: Jason Ekstrand <jason@jlekstrand.net> Acked-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/10582>
This commit is contained in:
Alyssa Rosenzweig
Alyssa Rosenzweig
parent
972409dacb
commit
50b5c94885
1 changed files with 185 additions and 0 deletions
185
src/asahi/compiler/agx_opcodes.py
Normal file
185
src/asahi/compiler/agx_opcodes.py
Normal file
|
|
@ -0,0 +1,185 @@
|
|||
"""
|
||||
Copyright (C) 2021 Alyssa Rosenzweig <alyssa@rosenzweig.io>
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a
|
||||
copy of this software and associated documentation files (the "Software"),
|
||||
to deal in the Software without restriction, including without limitation
|
||||
the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||
and/or sell copies of the Software, and to permit persons to whom the
|
||||
Software is furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice (including the next
|
||||
paragraph) shall be included in all copies or substantial portions of the
|
||||
Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
"""
|
||||
|
||||
opcodes = {}
|
||||
immediates = {}
|
||||
|
||||
class Opcode(object):
|
||||
def __init__(self, name, dests, srcs, imms, is_float, can_eliminate, encoding_16, encoding_32):
|
||||
self.name = name
|
||||
self.dests = dests
|
||||
self.srcs = srcs
|
||||
self.imms = imms
|
||||
self.is_float = is_float
|
||||
self.can_eliminate = can_eliminate
|
||||
self.encoding_16 = encoding_16
|
||||
self.encoding_32 = encoding_32
|
||||
|
||||
class Immediate(object):
|
||||
def __init__(self, name, ctype):
|
||||
self.name = name
|
||||
self.ctype = ctype
|
||||
|
||||
class Encoding(object):
|
||||
def __init__(self, description):
|
||||
(exact, mask, length_short, length_long) = description
|
||||
|
||||
# Convenience
|
||||
if length_long is None:
|
||||
length_long = length_short
|
||||
|
||||
self.exact = exact
|
||||
self.mask = mask
|
||||
self.length_short = length_short
|
||||
self.extensible = length_short != length_long
|
||||
|
||||
if self.extensible:
|
||||
assert(length_long == length_short + (4 if length_short > 8 else 2))
|
||||
|
||||
def op(name, encoding_32, dests = 1, srcs = 0, imms = [], is_float = False, can_eliminate = True, encoding_16 = None):
|
||||
encoding_16 = Encoding(encoding_16) if encoding_16 is not None else None
|
||||
encoding_32 = Encoding(encoding_32) if encoding_32 is not None else None
|
||||
|
||||
opcodes[name] = Opcode(name, dests, srcs, imms, is_float, can_eliminate, encoding_16, encoding_32)
|
||||
|
||||
def immediate(name, ctype = "uint32_t"):
|
||||
imm = Immediate(name, ctype)
|
||||
immediates[name] = imm
|
||||
return imm
|
||||
|
||||
L = (1 << 15)
|
||||
_ = None
|
||||
|
||||
FORMAT = immediate("format", "enum agx_format")
|
||||
IMM = immediate("imm")
|
||||
WRITEOUT = immediate("writeout")
|
||||
INDEX = immediate("index")
|
||||
COMPONENT = immediate("component")
|
||||
CHANNELS = immediate("channels")
|
||||
TRUTH_TABLE = immediate("truth_table")
|
||||
ROUND = immediate("round")
|
||||
SHIFT = immediate("shift")
|
||||
MASK = immediate("mask")
|
||||
BFI_MASK = immediate("bfi_mask")
|
||||
LOD_MODE = immediate("lod_mode", "enum agx_lod_mode")
|
||||
DIM = immediate("dim", "enum agx_dim")
|
||||
SCOREBOARD = immediate("scoreboard")
|
||||
ICOND = immediate("icond")
|
||||
FCOND = immediate("fcond")
|
||||
|
||||
FUNOP = lambda x: (x << 28)
|
||||
FUNOP_MASK = FUNOP((1 << 14) - 1)
|
||||
|
||||
def funop(name, opcode):
|
||||
op(name, (0x0A | L | (opcode << 28),
|
||||
0x3F | L | (((1 << 14) - 1) << 28), 6, _),
|
||||
srcs = 1, is_float = True)
|
||||
|
||||
# Listing of opcodes
|
||||
funop("floor", 0b000000)
|
||||
funop("srsqrt", 0b000001)
|
||||
funop("dfdx", 0b000100)
|
||||
funop("dfdy", 0b000110)
|
||||
funop("rcp", 0b001000)
|
||||
funop("rsqrt", 0b001001)
|
||||
funop("sin_pt_1", 0b001010)
|
||||
funop("log2", 0b001100)
|
||||
funop("exp2", 0b001101)
|
||||
funop("sin_pt_2", 0b001110)
|
||||
funop("ceil", 0b010000)
|
||||
funop("trunc", 0b100000)
|
||||
funop("roundeven", 0b110000)
|
||||
|
||||
op("fadd",
|
||||
encoding_16 = (0x26 | L, 0x3F | L, 6, _),
|
||||
encoding_32 = (0x2A | L, 0x3F | L, 6, _),
|
||||
srcs = 2, is_float = True)
|
||||
|
||||
op("fma",
|
||||
encoding_16 = (0x36, 0x3F, 6, 8),
|
||||
encoding_32 = (0x3A, 0x3F, 6, 8),
|
||||
srcs = 3, is_float = True)
|
||||
|
||||
op("fmul",
|
||||
encoding_16 = ((0x16 | L), (0x3F | L), 6, _),
|
||||
encoding_32 = ((0x1A | L), (0x3F | L), 6, _),
|
||||
srcs = 2, is_float = True)
|
||||
|
||||
op("mov_imm",
|
||||
encoding_32 = (0x62, 0xFF, 6, 8),
|
||||
encoding_16 = (0x62, 0xFF, 4, 6),
|
||||
imms = [IMM])
|
||||
|
||||
op("iadd",
|
||||
encoding_32 = (0x0E, 0x3F | L, 8, _),
|
||||
srcs = 2, imms = [SHIFT])
|
||||
|
||||
op("imad",
|
||||
encoding_32 = (0x1E, 0x3F | L, 8, _),
|
||||
srcs = 3, imms = [SHIFT])
|
||||
|
||||
op("bfi",
|
||||
encoding_32 = (0x2E, 0x7F | (0x3 << 26), 8, _),
|
||||
srcs = 3, imms = [BFI_MASK])
|
||||
|
||||
op("bfeil",
|
||||
encoding_32 = (0x2E | L, 0x7F | L | (0x3 << 26), 8, _),
|
||||
srcs = 3, imms = [BFI_MASK])
|
||||
|
||||
op("asr",
|
||||
encoding_32 = (0x2E | L | (0x1 << 26), 0x7F | L | (0x3 << 26), 8, _),
|
||||
srcs = 2)
|
||||
|
||||
op("icmpsel",
|
||||
encoding_32 = (0x12, 0x7F, 8, 10),
|
||||
srcs = 4, imms = [ICOND])
|
||||
|
||||
op("fcmpsel",
|
||||
encoding_32 = (0x02, 0x7F, 8, 10),
|
||||
srcs = 4, imms = [FCOND])
|
||||
|
||||
# sources are coordinates, LOD, texture, sampler, offset
|
||||
# TODO: anything else?
|
||||
op("texture_sample",
|
||||
encoding_32 = (0x32, 0x7F, 8, 10), # XXX WRONG SIZE
|
||||
srcs = 5, imms = [DIM, LOD_MODE, MASK, SCOREBOARD])
|
||||
|
||||
# sources are base, index
|
||||
op("device_load",
|
||||
encoding_32 = (0x05, 0x7F, 6, 8),
|
||||
srcs = 2, imms = [FORMAT, MASK, SCOREBOARD])
|
||||
|
||||
op("wait", (0x38, 0xFF, 2, _), dests = 0,
|
||||
can_eliminate = False, imms = [SCOREBOARD])
|
||||
|
||||
op("bitop", (0x7E, 0x7F, 6, _), srcs = 2, imms = [TRUTH_TABLE])
|
||||
op("blend", (0x09, 0x7F, 8, _), dests = 0, srcs = 1, imms = [FORMAT], can_eliminate = False)
|
||||
op("convert", (0x3E | L, 0x7F | L | (0x3 << 38), 6, _), srcs = 2, imms = [ROUND])
|
||||
op("ld_vary", (0x21, 0x3F, 8, _), srcs = 1, imms = [CHANNELS])
|
||||
op("st_vary", None, dests = 0, srcs = 2, can_eliminate = False)
|
||||
op("stop", (0x88, 0xFFFF, 2, _), dests = 0, can_eliminate = False)
|
||||
op("trap", (0x08, 0xFFFF, 2, _), dests = 0, can_eliminate = False)
|
||||
op("writeout", (0x48, 0xFF, 4, _), dests = 0, imms = [WRITEOUT], can_eliminate = False)
|
||||
|
||||
op("p_combine", _, srcs = 4)
|
||||
op("p_extract", _, srcs = 1, imms = [COMPONENT])
|
||||
Loading…
Add table
Reference in a new issue