Skip to content

Commit ee8cb4c

Browse files
Add AMDGPU methods to documentation
- Added AMDGPUOffloadLUFactorization and AMDGPUOffloadQRFactorization to solver docs - Added example usage in GPU tutorial section - Documented requirement for AMDGPU.jl package 🤖 Generated with Claude Code Co-Authored-By: Claude <[email protected]>
1 parent 9400fb7 commit ee8cb4c

File tree

2 files changed

+24
-1
lines changed

2 files changed

+24
-1
lines changed

docs/src/solvers/solvers.md

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -238,6 +238,19 @@ The following are non-standard GPU factorization routines.
238238
CudaOffloadFactorization
239239
```
240240

241+
### AMDGPU.jl
242+
243+
The following are GPU factorization routines for AMD GPUs using the ROCm stack.
244+
245+
!!! note
246+
247+
Using these solvers requires adding the package AMDGPU.jl, i.e. `using AMDGPU`
248+
249+
```@docs
250+
AMDGPUOffloadLUFactorization
251+
AMDGPUOffloadQRFactorization
252+
```
253+
241254
### CUSOLVERRF.jl
242255

243256
!!! note

docs/src/tutorials/gpu.md

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,21 @@ sol.u
3939
This computation can be moved to the GPU by the following:
4040

4141
```julia
42-
using CUDA # Add the GPU library
42+
using CUDA # Add the GPU library for NVIDIA GPUs
4343
sol = LS.solve(prob, LS.CudaOffloadFactorization())
4444
sol.u
4545
```
4646

47+
For AMD GPUs, you can use the AMDGPU.jl package:
48+
49+
```julia
50+
using AMDGPU # Add the GPU library for AMD GPUs
51+
sol = LS.solve(prob, LS.AMDGPUOffloadLUFactorization()) # LU factorization
52+
# or
53+
sol = LS.solve(prob, LS.AMDGPUOffloadQRFactorization()) # QR factorization
54+
sol.u
55+
```
56+
4757
## GPUArray Interface
4858

4959
For more manual control over the factorization setup, you can use the

0 commit comments

Comments
 (0)