Table of Contents
The GCN3 GPU is a model that simulates a GPU at the ISA level, as opposed to the intermediate language level. This page will give you a general overview of how to use this model, the software stack the model uses, and provide resources that detail the model and how it is implemented.
Currently, the GCN3 GPU model in gem5 is supported on the stable and develop branch.
The gem5 repository comes with a dockerfile located in util/dockerfiles/gcn-gpu/
. This dockerfile contains the drivers and libraries needed to run the GPU model. A pre-built version of the docker image is hosted at gcr.io/gem5-test/gcn-gpu:v22-0
.
The gem5-resources repository also comes with a number of sample applications that can be used to verify that the model runs correctly. We recommend users start with square, as it is a simple, heavily tested application that should run relatively quickly.
The docker image can either be built or pulled from gcr.io.
To build the docker image from source:
# Working directory: gem5/util/dockerfiles/gcn-gpu docker build -t <image_name> .
To pull the pre-built docker image (Note the v22-0
tag, to get the correct image for this release):
docker pull gcr.io/gem5-test/gcn-gpu:v22-0
You can also put gcr.io/gem5-test/gcn-gpu:v22-0
as the image in the docker run command without pulling beforehand and it will be pulled automatically.
See square in gem5 resources for an example of how to build gem5 in the docker. Note: these directions assume you are pulling the latest image automatically.
See gem5 resources for examples of how to build and run GPU applications in the docker.
The GCN3 model was designed with enough fidelity to not require an emulated runtime. Instead, the GCN3 model uses the Radeon Open Compute platform (ROCm). ROCm is an open platform from AMD that implements Heterogeneous Systems Architecture (HSA) principles. More information about the HSA standard can be found on the HSA Foundation's website. More information about ROCm can be found on the ROCm website
The model currently only works with system-call emulation (SE) mode, therefore all kernel level driver functionality is modeled entirely within the SE mode layer of gem5. In particular, the emulated GPU driver supports the necessary ioctl()
commands it receives from the userspace code. The source for the emulated GPU driver can be found in:
The GPU compute driver: src/gpu-compute/gpu_compute_driver.[hh|cc]
The HSA device driver: src/dev/hsa/hsa_driver.[hh|cc]
The HSA driver code models the basic functionality for an HSA agent, which is any device that can be targeted by the HSA runtime and accepts Architected Query Language (AQL) packets. AQL packets are a standard format for all HSA agents, and are used primarily to initiate kernel launches on the GPU. The base HSADriver
class holds a pointer to the HSA packet processor for the device, and defines the interface for any HSA device. An HSA agent does not have to be a GPU, it could be a generic accelerator, CPU, NIC, etc.
The GPUComputeDriver
derives from HSADriver
and is a device-specific implementation of an HSADriver
. It provides the implementation for GPU-specific ioctl()
calls.
The src/dev/hsa/kfd_ioctl.h
header must match the kfd_ioctl.h
header that comes with ROCt. The emulated driver relies on that file to interpret the ioctl()
codes the thunk uses.
The GCN3 model supports ROCm version 4.0.
The following ROCm components are required:
The following additional components are used to build and run machine learning programs:
For information about installing these components locally, the commands in the GCN3 dockerfile (util/dockerfiles/gcn-gpu/
) can be followed on an Ubuntu 16 machine.
Describes the GCN3 model
Covers information about the GPU architecture, GCN3 ISA and HW-SW interfaces in gem5. Also provides an introduction to ROCm.
Contains further documentation about the ROCm stack, as well as programming guides for using ROCm.