Skip to content

ROCm Support

Jump to bottom
likelovewant edited this page Jul 24, 2024 · 6 revisions

This guide will walk you through building rocBLAS using the official ROCm documentation.

This guide is for users with AMD GPUs lacking official ROCm/HIP SDK support, or those wanting to enable HIP SDK support for hip sdk 5.7 and 6.1.2 on Windows for integrated AMD GPUs(iGPUs)."

If you already have the libraries, you can skip this section!

To build your own rocBLAS, follow this guide. A simplified version is available on GitHub: ROCm-HIP-SDK-Windows-Support.

Prerequisites: Ensure the following software is installed on your PC. python, git, and the HIP SDKare essential. The script rdeps.py will automatically download any missing dependencies when you run it.

Downloading the Source Code:

  1. rocBLAS: Download the latest version (https://github.com/ROCm/rocBLAS).

  2. Tensile: Download the appropriate version:(https://github.com/ROCm/Tensile)

Patching Tensile for ROCm (For Advanced Users, Not-a-must-Do)

These steps are necessary for specific configurations of ROCm and may not be required in all cases. If you had a optimized logic for you gpu arche or do the necessary edit as guide here,you may skip this steps.Especily build libs for xnack- features.

Determine Your ROCm Version:

  • ROCm 5.7.0: Follow the instructions for "For hip 5.7" below.
  • ROCm 6.1.2: Follow the instructions for "For hip 6.1.2" below.

Patches for Tensile:

For hip 5.7.0:

  1. Download Tensile-fix-fallback-arch-build.patch.

  2. Place the patch file in your Tensile folder (e.g., C:\ROCM\Tensile-rocm-5.7.0).

  3. Open a terminal within the Tensile folder.

  4. Apply the patch:

    git apply Tensile-fix-fallback-arch-build.patch
    • If nothing appears after applying, it's patched successfully. Otherwise, you may need to manually add the patch content to TensileCreateLibrary.py, you may also skip this steps if you have optimized logic available.

For hip 6.1.2:

  1. Download Tensile-fix-fallback-arch-build-hip-6.1.2.patch.

  2. Place the patch file in your Tensile folder (e.g., C:\ROCM\Tensile-rocm-6.1.2).

  3. Open a terminal within the Tensile folder.

  4. Apply the patch:

    git apply Tensile-fix-fallback-arch-build-hip-6.1.2.patch
    • If nothing appears after applying, it's patched successfully. Otherwise, you may need to manually add the patch content to TensileCreateLibrary.py.

( Skip this step for ROCm 6.1.2 )

Note: edit the line 41 in file rdeps.py for rocBLAS ,The old repo has an outdated vckpg, which will lead to fail build.update the vcpkg ,by replace with the following line

git clone -b 2024.02.14 https://github.com/microsoft/vcpkg

to udpate the vckpg version.

  • vcpkg Version: If your vcpkg version was built after April 2023, replace CMakeLists.txt in Tensile/tree/develop/Tensile/Source/lib/CMakeLists.txt with this version and place it in the same folder (e.g., rocm).

Build with rdeps and rmake:

  1. Navigate to the rocm/rocBLAS directory in your terminal.
  2. Run python rdeps.py. This script will configure your environment and download necessary packages.
    python rdeps.py

( using install.sh -d in linux , if you encounter any mistakes , try to google and fix with it or try it again ) after done . try next step

  1. After rdeps.py completes, run

python rmake.py -a "gfx1101;gfx1103" --lazy-library-loading--no-merge-architectures -t "C:\rocm\Tensile-rocm-5.7.0"

(adjust paths and architectures as needed).

Important:

  • Replace "gfx1101;gfx1103" with the correct GPU or APU architecture names for your system.Make sure sepearte with ";"if you have more than one arches build .
  • Make sure read the Editing Tensile/Common.py and blow before to build .
  • For ROCm 6.1.2, change the path to C:\rocm\Tensile-rocm-6.1.2.
  • The specific commands and patch files may vary depending on your setup and ROCm version.

After successfully building rocBLAS from source, you need to replace the default rocblas.dll with your compiled version for your HIP programs to utilize it. Here's how:

  1. Locate your Compiled Files:

    • rocblas.dll: Located in C:\ROCM\rocBLAS-rocm-5.7.0\build\release\staging\ (or a similar path based on your build location).
    • Tensile data files: Found within C:\ROCM\rocBLAS-rocm-5.7.0\build\release\Tensile\library\ (adjust the path if needed).

  2. Replace the Default rocBLAS:

    • Copy rocblas.dll to C:\Program Files\AMD\ROCm\5.7\bin. This is where the HIP SDK looks for it by default.( make sure to bakc up the origianl rocblas.dll )

  3. Place Tensile Data Files:

    • Navigate to C:\Program Files\AMD\ROCm\5.7\bin\rocblas\
    • Replace the library with new build ( back up the origianl library by rename to different name ,eg ,bklibrary). This is where you should place all the Tensile data files from your build directory.

  4. Test Your HIP Program:

    • Now, when you run your HIP program, it should use your newly compiled rocblas.dll and its associated Tensile data files.

Important Notes:

  • For ROCm 6.1.2, change the path to C:\Program Files\AMD\ROCm\6.1\bin\.
  • Always double-check the paths to ensure they match your installation configuration.
  • Make sure the ROCm version in the bin directory matches the version of rocBLAS you built.

Note: Editing Tensile/Common.py

This file contains general parameters used by the Tensile library. To ensure compatibility with your GPU, you need to update two specific settings.Update the value of " globalParameters["SupportedISA"]" and "CACHED_ASM_CAPS" with yourgpu ISA and info .and choose the simliar gpu achetecture. eg RND2 for gfx1031 ,RND2 for gfx1032, then copy and put below with your gpu number and others availble gpu data .For hip sdk 6.1.2 , CACHED_ASM_CAPS info move to tensile/AsmCaps.py, also edit architectureMap from line299 to 310 , add your arch infomation .map your arch information to correct logic file .however , some optimized logic don't exsit in the offoicial release. then we need to creat it.otherwilse ,it will creat a fallback no optimized rocblas and library.

Here's a step-by-step guide:

  1. Choose Your Architecture:

    • Select an existing architecture folder within rocBLAS\library\src\blas3\Tensile\Logic\asm_full (e.g., navi21). This will serve as a template for your new architecture.
    • Create a new folder with the name of your target architecture (e.g., navi22).

  2. Copy Files:

    • Copy all the files from your chosen template folder into your new architecture folder.

  3. Modify Files:

    • Open the copied files in a code editor (like VS Code or Visual Studio).
    • Search for instances of navi21 and replace them with navi22.
    • Update any gfx1030 references to gfx1031 (or your target GPU's identifier).
    • Find lines containing ISA: [10, 3, 0] and replace them with ISA: [10, 3, 1]. (Remember to adjust the ISA code according to your GPU)
    • "Rename all files within the new folder to reflect your architecture name (e.g., change 'navi21' to 'navi22'). You can use a file renaming tool like 'File Rename APP', a free application available in the Windows Store, for this task."
    • if build failed ,that's beacuse ROCm architectures have different capabilities. You need to ensure your rocblas is tailored to each architecture you're targeting:

      • gfx90c: Doesn't support 4x8II. Delete any logic or files related to 4x8II within the asm_full folder under rocBLAS\library\src\blas3\Tensile\Logic.

      • gfx1010: Doesn't support 8II. Do the same for files related to 8II in the asm_full folder.

    • Checking Logic Files: The "new named logic file" is likely a critical place where these operations are defined. Carefully review it and remove any unsupported calculations.

  4. Use Your New Architecture:

    • In Tensile/Common.py, update "CACHED_ASM_CAPS" or the relevant entries in architectureMap to reference your new navi22 folder.

Important Notes:

  • Carefully review the changes you make, as incorrect modifications can lead to errors.

(Skip this for HIP 5.7, Necessary for HIP 6.1.2)

Key Changes:

  • Search for gfx1030: Begin by searching within both the Tensile and rocBLAS folders for instances of gfx1030. This identifier represents a gfx1030 GPU architecture.
  • Replace with Your Target Architecture: Replace all occurrences of gfx1030 with the corresponding code for your desired GPU architecture (e.g., gfx1031).

Important Files to Modify:

  • Tensile: Within the Tensile folder, make changes to:

    • CMakeLists.txt: This file configures the build process and needs adjustments for new architectures.
    • AMDGPU.hpp: Defines the architecture-specific interface.
    • PlaceholderLibrary.hpp, Predicaters.hpp, OclUtiles.cpp: These files contain code related to specific functionalities, which might require modifications for your target GPU.

  • rocBLAS: In the rocBLAS folder:

    • CMakeLists.txt: Similar to Tensile, update this file for your new architecture.
    • handle.cpp, tensile_host.cpp, handle.hpp: These files are likely involved in communication and interactions between rocBLAS and the GPU.

Caution:

  • Modifying these core files can have unintended consequences.

Advanced Usage:

For maximum performance optimization, delve deeper into Tensile's logic files. Examples are provided in rocBLAS\library\src\blas3\Tensile\Logic\asm_full.

For truly optimized libraries, you'll need to fine-tune these logic files specifically for your target hardware.The Tensile Tuning Guide provides practical guidance and techniques for start this process. Keep in mind that the tuning process requires patience, time, and a willingness to delve into Tensile's inner workings.

More detail can be found in tuning , and tensile tuning .tex , A pdf version available in here

Please feel welcome to edit this post and contribute optimized logic links. Remember to carefully consider the impact of any edits or additions.

Clone this wiki locally