configs,dev-amdgpu: GPUFS MI200/gfx90a support
Add support for MI200-like device. This includes adding PCI IDs and new MMIOs for the device, a different MAP_PROCESS packet, and a different calculation for the number of VGPRs. Change-Id: I0fb7b3ad928826beaa5386d52a94ba504369cb0d Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/70317 Reviewed-by: Jason Lowe-Power <power.jg@gmail.com> Maintainer: Jason Lowe-Power <power.jg@gmail.com> Tested-by: kokoro <noreply+kokoro@google.com>
This commit is contained in:
@@ -132,8 +132,9 @@ def addRunFSOptions(parser):
|
||||
parser.add_argument(
|
||||
"--gpu-device",
|
||||
default="Vega10",
|
||||
choices=["Vega10", "MI100"],
|
||||
help="GPU model to run: Vega10 (gfx900) or MI100 (gfx908)",
|
||||
choices=["Vega10", "MI100", "MI200"],
|
||||
help="GPU model to run: Vega10 (gfx900), MI100 (gfx908), or "
|
||||
"MI200 (gfx90a)",
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -177,6 +177,10 @@ def connectGPU(system, args):
|
||||
system.pc.south_bridge.gpu.DeviceID = 0x738C
|
||||
system.pc.south_bridge.gpu.SubsystemVendorID = 0x1002
|
||||
system.pc.south_bridge.gpu.SubsystemID = 0x0C34
|
||||
elif args.gpu_device == "MI200":
|
||||
system.pc.south_bridge.gpu.DeviceID = 0x740F
|
||||
system.pc.south_bridge.gpu.SubsystemVendorID = 0x1002
|
||||
system.pc.south_bridge.gpu.SubsystemID = 0x0C34
|
||||
elif args.gpu_device == "Vega10":
|
||||
system.pc.south_bridge.gpu.DeviceID = 0x6863
|
||||
else:
|
||||
|
||||
@@ -152,6 +152,16 @@ def makeGpuFSSystem(args):
|
||||
0x7D000,
|
||||
]
|
||||
sdma_sizes = [0x1000] * 8
|
||||
elif args.gpu_device == "MI200":
|
||||
num_sdmas = 5
|
||||
sdma_bases = [
|
||||
0x4980,
|
||||
0x6180,
|
||||
0x78000,
|
||||
0x79000,
|
||||
0x7A000,
|
||||
]
|
||||
sdma_sizes = [0x1000] * 5
|
||||
else:
|
||||
m5.util.panic(f"Unknown GPU device {args.gpu_device}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user