ROCm for LoongArch - rocm-loongarch

🇨🇳

Welcome to the rocm-loongarch repository, a community-driven project hosted under the loongarch-moe organization. This repository contains patches and software packages for adapting the ROCm software stack to support GPUs on the LoongArch architecture. The work primarily focuses on supporting AMDGPU and enabling high-performance GPU computing on LoongArch.

Project Overview

The rocm-loongarch project provides the following stages of adaptation for running the ROCm software stack on LoongArch-based systems:

1. Kernel Stage

• Enabling HSA_AMD driver support
• Switching to 4KB page size

2. Stage 1 - Environment Setup

Basic ROCm infrastructure packages:

• rocm-core
• rocm-llvm-core
• rocm-llvm-clang
• rocm-llvm-compiler-rt
• rocm-llvm-lld
• rocm-llvm-extra

3. Stage 2 - ROCm Management and Runtime

• rocm-rocprofiler-register
• rocm-rocr-runtime
• rocminfo
• rocm-cmake
• rocm-clr
• rocm_bandwidth_test
• rocm-half
• rocm_smi_lib
• rocm-HIPIFY
• rocm-rocPRIM
• rocm-hipCUB
• mscclpp

4. Stage 3 - Additional ROCm Dependencies

• rocm-llvm-openmp
• rocm-rocMLIR
• rocm-llvm-mlir
• rocm-llvm-flang
• rocm-hipfort
• rocm-rocJPEG
• rocm-pgmath
• rocm-flang-classic
• rocm-rocDecode

5. Stage 4 - Algorithm Libraries

• rocm-hipBLAS-common
• rocm-hipBLASLt
• rocm-rocBLAS
• rocm-rocFFT
• rocm-rocRAND
• rocm-rccl
• rocm-rocSPARSE
• rocm-hipFFT
• rocm-hipSPARSE
• rocm-rocSOLVER
• rocm-hipBLAS
• rocm-hipRAND
• rocm-hipSPARSELt

6. Stage 5 - Advanced Algorithm Libraries

• rocm-hipSOLVER
• rocm-rocThrust
• rocm-rocALUTION
• rocm-composable_kernel
• rocm-MIOpen
• rocm-ROCdbgapi

7. Stage 6 - Applications

• stable-diffusion.cpp
• pytorch-rocm (in progress)
• pytorch-cpu
• llama.cpp
• cupy
• aotriton

What is ROCm

For a detailed introduction to ROCm, including its components and architecture, you can refer to the AMD ROCm documentation.

Kernel and RDNA Version Compatibility

Here is the recommended relationship between kernel versions and RDNA generations:

• Kernel version 6.14 + rx9070_rdna4_fpu.patch: 4K page size support for RDNA4 serial.
• Kernel version 6.8 and above: 4K page size support for RDNA2/3 serial.
• Kernel version below 6.8: 4K/16K page size support for RDNA2 serial.

Support GPU ISA

GPU Architecture	Shader ISA	Support	Need PCIe Atomics?	Need HSA_OVERRIDE?
GCN 4.0	gfx80x	No
GCN 5.0	gfx900/1	Unknown
GCN 5.1	gfx906/7	Unknown
CDNA 1.0	gfx908	Unknown
CDNA 2.0	gfx90a	Unknown
CDNA 3.0	gfx940/2	Unknown
RDNA 1.0	gfx101x	Unknown
RDNA 2.0	gfx1030	Yes	No
RDNA 2.0	gfx103x	Yes	No	10.3.0
RDNA 3.0	gfx110x	Yes	No
RDNA 3.5	gfx115x	No(iGPU only)
RDNA 4.0	gfx120x	Unknown

Releases

For the latest releases of the software and packages, please visit the Releases page.

Contributing

We welcome contributions to this project. If you're interested in improving or extending the support for ROCm on LoongArch, please feel free to fork the repository, submit issues, and create pull requests.

Contribution Guidelines

• Ensure that your changes are well-tested and documented.
• Follow the standard coding conventions for ROCm and LoongArch development.

Contact Us

For any questions or issues, feel free to open an issue in the repository, or reach out to the maintainers via GitHub Discussions.