diff --git a/util/dockerfiles/gcn-gpu/Dockerfile b/util/dockerfiles/gcn-gpu/Dockerfile index 9aead08c43..6ccb150c12 100644 --- a/util/dockerfiles/gcn-gpu/Dockerfile +++ b/util/dockerfiles/gcn-gpu/Dockerfile @@ -104,7 +104,7 @@ RUN apt install rocm-cmake RUN git clone -b rocm-4.0.0 \ https://github.com/ROCmSoftwarePlatform/rocBLAS.git && mkdir rocBLAS/build -ENV HCC_AMDGPU_TARGET=gfx801,gfx803,gfx900,gfx902 +ENV HCC_AMDGPU_TARGET=gfx900,gfx902 WORKDIR rocBLAS # rocBLAS needs to be built from source otherwise certain gfx versions get errors in HIP # about there being no GPU binary available @@ -125,7 +125,7 @@ RUN mkdir -p /root/.cache/miopen/2.9.0.8252-rocm-rel-4.0-26-64506314 && \ # Add commands from halofinder Dockerfile RUN apt-get update && apt-get -y install libopenmpi-dev libomp-dev -ENV HCC_AMDGPU_TARGET="gfx801,gfx803,gfx900" +ENV HCC_AMDGPU_TARGET="gfx900" ENV HIPCC_BIN=/opt/rocm/bin ENV MPI_INCLUDE=/usr/lib/x86_64-linux-gnu/openmpi/include diff --git a/util/dockerfiles/gcn-gpu/README.md b/util/dockerfiles/gcn-gpu/README.md index 0764cad415..2c5a21e7df 100644 --- a/util/dockerfiles/gcn-gpu/README.md +++ b/util/dockerfiles/gcn-gpu/README.md @@ -1,5 +1,5 @@ -## gcn3-gpu dockerfile -This dockerfile contains all the dependences necessary to run GPU applications in gem5 using the gcn3 APU model +## gcn-gpu dockerfile +This dockerfile contains all the dependences necessary to run GPU applications in gem5 using the gcn5 (Vega) APU model ### Building the image ``` @@ -9,14 +9,14 @@ docker build -t . ### Building gem5 using the image The following command assumes the gem5 directory is a subdirectory of your current directory ``` -docker run --rm -v $PWD/gem5:/gem5 -w /gem5 scons -sQ -j$(nproc) build/GCN3_X86/gem5.opt +docker run --rm -v $PWD/gem5:/gem5 -w /gem5 scons -sQ -j$(nproc) build/VEGA_X86/gem5.opt ``` ### Test gem5 using a prebuilt application ``` wget http://dist.gem5.org/dist/current/test-progs/hip_sample_bins/MatrixTranspose docker run --rm -v $PWD/MatrixTranspose:/MatrixTranspose -v $PWD/public_gem5:/gem5 -w /gem5 \ - build/GCN3_X86/gem5.opt configs/example/apu_se.py -n2 --benchmark-root=/ -cMatrixTranspose + build/VEGA_X86/gem5.opt configs/example/apu_se.py -n2 --benchmark-root=/ -cMatrixTranspose ``` ### Notes