xuchen-amd
156 regels
4.2 KiB
YAML
156 regels
4.2 KiB
YAML
# --------------------------------------------------------------------------------
|
|
#
|
|
# This yaml file tracks MI gpu spec in a tree structure.
|
|
#
|
|
# *It is important to note that this file only tracks the common specs for MI GPU series.*
|
|
# *For example, the CU #s are based on information retrieved from other tools.*
|
|
# **
|
|
#
|
|
# MI GPUs
|
|
# |-- series: the specific MI series; mi50, mi100, mi200, mi300
|
|
# |-- architecture: currently, only mi300 gpus hold different architectures
|
|
# |-- gpu model
|
|
# |-- chip_ids: chip id is specific to the environment the gpu is being used on
|
|
# |-- partition_mode
|
|
# | -- compute partition mode
|
|
# | -- memory partition mode
|
|
#
|
|
# --------------------------------------------------------------------------------
|
|
|
|
mi_gpu_spec:
|
|
- gpu_series: mi50
|
|
gpu_archs:
|
|
- gpu_arch: gfx906
|
|
models:
|
|
- gpu_model: mi50
|
|
- gpu_model: mi60
|
|
|
|
- gpu_series: mi100
|
|
gpu_archs:
|
|
- gpu_arch: gfx908
|
|
models:
|
|
- gpu_model: mi100
|
|
chip_ids:
|
|
physical: 29580
|
|
|
|
- gpu_series: mi200
|
|
gpu_archs:
|
|
- gpu_arch: gfx90a
|
|
models:
|
|
- gpu_model: mi210
|
|
chip_ids:
|
|
physical: 29711
|
|
- gpu_model: mi250
|
|
chip_ids:
|
|
physical: 29708
|
|
- gpu_model: mi250x
|
|
chip_ids:
|
|
physical: 29704
|
|
- gpu_model: mi250
|
|
- gpu_model: mi250x
|
|
|
|
- gpu_series: mi300
|
|
gpu_archs:
|
|
- gpu_arch: gfx940
|
|
models:
|
|
- gpu_model: mi300a_a0
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 6
|
|
tpx: 2
|
|
memory_partition_mode:
|
|
nps4: [tpx]
|
|
nps1: [spx, tpx]
|
|
|
|
- gpu_arch: gfx941
|
|
models:
|
|
- gpu_model: mi300x_a0
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 8
|
|
dpx: 4
|
|
qpx: 2
|
|
cpx: 1
|
|
memory_partition_mode:
|
|
nps4: [qpx, cpx]
|
|
nps1: [spx, qpx, cpx]
|
|
|
|
- gpu_arch: gfx942
|
|
models:
|
|
- gpu_model: mi300a_a1
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 6
|
|
tpx: 2
|
|
memory_partition_mode:
|
|
nps4: [tpx]
|
|
nps1: [spx, tpx]
|
|
chip_ids:
|
|
physical: 29856
|
|
virtual: 29876
|
|
|
|
- gpu_model: mi300x_a1
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 8
|
|
dpx: 4
|
|
qpx: 2
|
|
cpx: 1
|
|
memory_partition_mode:
|
|
nps4: [qpx, cpx]
|
|
nps1: [spx, qpx, cpx]
|
|
chip_ids:
|
|
physical: 29857
|
|
virtual: 29877
|
|
|
|
- gpu_model: mi308x
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 4
|
|
dpx: 2
|
|
cpx: 1
|
|
memory_partition_mode:
|
|
nps4: [cpx]
|
|
nps1: [spx, dpx, cpx]
|
|
chip_ids:
|
|
physical: 29858
|
|
virtual: 29878
|
|
|
|
- gpu_model: mi325x
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 8
|
|
dpx: 4
|
|
qpx: 2
|
|
cpx: 1
|
|
memory_partition_mode:
|
|
nps4: [qpx, cpx]
|
|
nps1: [spx, qpx, cpx]
|
|
chip_ids:
|
|
physical: 29861
|
|
virtual: 29881
|
|
|
|
- gpu_series: mi350
|
|
gpu_archs:
|
|
- gpu_arch: gfx950
|
|
models:
|
|
- gpu_model: mi350
|
|
partition_mode:
|
|
compute_partition_mode:
|
|
num_xcds:
|
|
spx: 8
|
|
dpx: 4
|
|
qpx: 2
|
|
cpx: 1
|
|
memory_partition_mode:
|
|
nps1: [spx, dpx, qpx, cpx]
|
|
nps4: [qpx, cpx]
|
|
chip_ids:
|
|
physical: 30112
|