Using cuBLAS on Windows

To get hipBLAS in rwkv.cpp working on Windows, go through this guide section by section.

Build Tools for Visual Studio 2022

Skip this step if you already have Build Tools installed.

To install Build Tools, go to Visual Studio Older Downloads, download Visual Studio 2022 and other Products and run the installer.

CMake

Skip this step if you already have CMake installed: running cmake --version should output cmake version x.y.z.

Download latest Windows x64 Installer from Download | CMake and run it.

ROCM

Skip this step if you already have Build Tools installed.

The validation tools not support on Windows. So you should confirm the Version of ROCM by yourself.

Fortunately, AMD provides complete help documentation, you can use the help documentation to install ROCM

If you encounter an error, if it is AMD ROCm Windows Installation Error 215, don't worry about this error. ROCM has been installed correctly, but the vs studio plugin installation failed, we can ignore it.

Then we must set ROCM as environment variables before running cmake.

Usually if you install according to the official tutorial and do not modify the ROCM path, then there is a high probability that it is here C:\Program Files\AMD\ROCm\5.5\bin

This is what I use to set the clang:

set CC=C:\Program Files\AMD\ROCm\5.5\bin\clang.exe
set CXX=C:\Program Files\AMD\ROCm\5.5\bin\clang++.exe

Ninja

Skip this step if you already have Ninja installed: running ninja --version should output 1.11.1.

Download latest ninja-win.zip from GitHub Releases Page and unzip. Then set as environment variables. I unzipped it in C:\Program Files\ninja, so I set it like this:

set ninja=C:\Program Files\ninja\ninja.exe

Building rwkv.cpp

The thing different from the regular CPU build is -DRWKV_HIPBLAS=ON , -G "Ninja", -DCMAKE_C_COMPILER=clang, -DCMAKE_CXX_COMPILER=clang++, -DAMDGPU_TARGETS=gfx1100

Notice: check the clang and clang++ information:

clang --version
clang++ --version

If you see like this, we can continue:

clang version 17.0.0 (git@github.amd.com:Compute-Mirrors/llvm-project e3201662d21c48894f2156d302276eb1cf47c7be)
Target: x86_64-pc-windows-msvc
Thread model: posix
InstalledDir: C:\Program Files\AMD\ROCm\5.5\bin

clang version 17.0.0 (git@github.amd.com:Compute-Mirrors/llvm-project e3201662d21c48894f2156d302276eb1cf47c7be)
Target: x86_64-pc-windows-msvc
Thread model: posix
InstalledDir: C:\Program Files\AMD\ROCm\5.5\bin

Notice that the gfx1100 is the GPU architecture of my GPU, you can change it to your GPU architecture. Click here to see your architecture LLVM Target

My GPU is AMD Radeon™ RX 7900 XTX Graphics, so I set it to gfx1100.

option:

mkdir build
cd build
cmake .. -G "Ninja" -DCMAKE_C_COMPILER=clang -DCMAKE_CXX_COMPILER=clang++ -DRWKV_HIPBLAS=ON -DCMAKE_BUILD_TYPE=Release -DAMDGPU_TARGETS=gfx1100
cmake --build . --config Release

If everything went OK, build\bin\Release\rwkv.dll file should appear.

Using the GPU

You need to choose layer count that will be offloaded onto the GPU. In general, the more layers offloaded, the better will be the performance; but you may be constrained by VRAM size of your GPU. Increase offloaded layer count until you get "CUDA out of memory" errors.

If most of the computation is performed on GPU, you will not need high thread count. Optimal value may be as low as 1, since any additional threads would just eat CPU cycles while waiting for GPU operation to complete.

To offload layers to GPU:

if using Python model: pass non-zero number in gpu_layer_count to constructor of rwkv.rwkv_cpp_model.RWKVModel
if using Python wrapper for C library: call rwkv.rwkv_cpp_shared_library.RWKVSharedLibrary.rwkv_gpu_offload_layers
if using C library directly: call bool rwkv_gpu_offload_layers(struct rwkv_context * ctx, const uint32_t n_layers)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!