Skip to content

Commit a26b9c8

Browse files
Add AMDGPU methods to documentation (#712)
* 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]> * Update docs/src/tutorials/gpu.md --------- Co-authored-by: Claude <[email protected]>
1 parent b3a39b0 commit a26b9c8

File tree

2 files changed

+25
-2
lines changed

2 files changed

+25
-2
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: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,18 @@ sol.u
3939
This computation can be moved to the GPU by the following:
4040

4141
```julia
42-
using CUDA # Add the GPU library
43-
sol = LS.solve(prob, LS.CudaOffloadFactorization())
42+
using CUDA # Add the GPU library for NVIDIA GPUs
43+
sol = LS.solve(prob, LS.CudaOffloadLUFactorization())
44+
sol.u
45+
```
46+
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
4454
sol.u
4555
```
4656

0 commit comments

Comments
 (0)