Skip to content

Commit 6199e67

Browse files
committed
Update README_AMD_WINDOWS.md
1 parent 4b7674c commit 6199e67

File tree

1 file changed

+7
-30
lines changed

1 file changed

+7
-30
lines changed

README_AMD_WINDOWS.md

Lines changed: 7 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ This guide explains how to build and run TurboDiffusion on Windows with AMD GPUs
66
77
## Supported Hardware
88

9-
TurboDiffusion on Windows has been tested with RDNA3/RDNA3.5 GPUs (gfx1100, gfx1101, gfx1102, gfx1103, gfx1151).
9+
TurboDiffusion on Windows has been tested with RDNA3/RDNA3.5 GPUs (gfx1100, gfx1101, gfx1102, gfx1151).
1010

1111
## Prerequisites
1212

@@ -53,28 +53,10 @@ rocm-sdk init
5353

5454
#### Install Triton with AMD Windows Support
5555

56-
Install triton-windows and apply the AMD Windows support patches:
57-
5856
```powershell
5957
pip install triton-windows
6058
```
6159

62-
**Temporary Fix:** Until [PR #179](https://github.com/woct0rdho/triton-windows/pull/179) is merged, you need to manually copy the patched files into your venv:
63-
64-
1. Clone the PR branch:
65-
```powershell
66-
git clone --branch jam/windows_amd https://github.com/woct0rdho/triton-windows.git triton-windows-patch
67-
```
68-
69-
2. Copy the patched files to your venv:
70-
```powershell
71-
$TRITON_AMD = ".\venv\Lib\site-packages\triton\backends\amd"
72-
Copy-Item "triton-windows-patch\third_party\amd\backend\driver.py" "$TRITON_AMD\driver.py" -Force
73-
Copy-Item "triton-windows-patch\third_party\amd\backend\driver.c" "$TRITON_AMD\driver.c" -Force
74-
Copy-Item "triton-windows-patch\third_party\amd\backend\compiler.py" "$TRITON_AMD\compiler.py" -Force
75-
Copy-Item "triton-windows-patch\python\triton\runtime\build.py" ".\venv\Lib\site-packages\triton\runtime\build.py" -Force
76-
```
77-
7860
### 2. Set Environment Variables
7961

8062
Open a PowerShell terminal and run:
@@ -105,25 +87,20 @@ $env:TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL = "1"
10587
$env:PYTHONPATH = "turbodiffusion"
10688
```
10789

108-
### 3. Install Dependencies
109-
110-
```powershell
111-
pip install -r requirements.txt
112-
```
113-
114-
### 4. Build and Install TurboDiffusion
90+
### 3. Build and Install TurboDiffusion
11591

11692
```powershell
11793
cd <path_to_turbodiffusion>
11894
pip install --no-build-isolation -e .
11995
```
12096

121-
### 5. Install SpargeAttn (Optional, for sparse attention)
97+
### 4. Install SpargeAttn (Optional, for sparse attention)
12298

123-
If you want to use sparse attention with TurboDiffusion:
99+
If you want to use sparse attention with TurboDiffusion, clone the AMD Windows fork:
124100

125101
```powershell
126-
cd <path_to_spargeattn>
102+
git clone --branch jam/amd_windows https://github.com/jammm/SpargeAttn.git
103+
cd SpargeAttn
127104
pip install --no-build-isolation -v .
128105
```
129106

@@ -187,7 +164,7 @@ cmd /c '"C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Bu
187164

188165
### "PermissionError" when compiling Triton kernels
189166

190-
This is a known Windows issue with temp file handling. Make sure you've applied the AMD Windows support patches from [PR #179](https://github.com/woct0rdho/triton-windows/pull/179) as described in the installation steps above.
167+
This is a known Windows issue with temp file handling. Make sure you're using the latest `triton-windows` package (`pip install --upgrade triton-windows`).
191168

192169
### "flash_attn is not installed" warning
193170

0 commit comments

Comments
 (0)