Skip to content

Commit d60a3cb

Browse files
committed
Add PyTorch, TensorFlow, and GROMACS case studies to ROCm overview
1 parent 893502a commit d60a3cb

File tree

1 file changed

+84
-19
lines changed

1 file changed

+84
-19
lines changed

docs/site_specific_config/rocm.md

Lines changed: 84 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
# Overview of ROCm Ecosystem (v6.4.1-20250616)
1+
# Overview of ROCm Ecosystem (v6.4.1-20250620)
22

33
!!! warning "Work-in-progress"
44
This document is a work-in-progress.
55
It may still contain inaccuracies or mistakes.
66

77
This overview is being created in the context of adding support for ROCm to EESSI, the European Environment for Scientific Software Installations (<https://eessi.io>).
88

9-
*Last update: 16 Jun 2025*
9+
*Last update: 20 Jun 2025*
1010

1111
[Jump to Overview](#Introduction) | [Jump to ABC](#ABC-of-ROCm) | [Jump to Changelog](#Changelog)
1212

@@ -331,17 +331,20 @@ The "roc" variants like rocFFT are AMD's native implementations optimized specif
331331

332332
### ML/DL Frameworks
333333

334+
* MIGraphX: Graph optimization engine ([Github](https://github.com/ROCm/AMDMIGraphX))
334335
* MIOpen: Deep learning primitives library ([Github](https://github.com/ROCm/MIOpen))
335336
* ROCm PyTorch: PyTorch support for AMD GPUs ([Github](https://github.com/ROCm/pytorch))
336337
* ROCm TensorFlow: TensorFlow support for AMD GPUs ([Github](https://github.com/ROCm/tensorflow-upstream))
337338

338339
### Communication Libraries
339340

340341
* RCCL: Communication library for multi-GPU/multi-node training ([Github](https://github.com/ROCm/rccl))
342+
* rocPRIM: Provides parallel primitives used to develop GPU-accelerated code ([Github](https://github.com/ROCm/rocPRIM))
341343

342344
### Marshalling Libraries
343345

344346
* hipBLAS ([Github](https://github.com/ROCm/hipBLAS))
347+
* hipCUB ([Github](https://github.com/ROCm/hipCUB))
345348
* hipFFT ([Github](https://github.com/ROCm/hipFFT))
346349
* hipRAND ([Github](https://github.com/ROCm/hipRAND))
347350
* hipSOLVER ([Github](https://github.com/ROCm/hipSOLVER))
@@ -358,50 +361,107 @@ graph LR;
358361
roctracer[ROC Tracer]
359362
end
360363
subgraph Libraries and Frameworks
364+
hipblaslt[hipBLASLt]
365+
hipsparselt[hipSPARSELt]
366+
rocblas[rocBLAS]
361367
rocfft[rocFFT]
362368
rocrand[rocRAND]
363-
rocblas[rocBLAS]
364-
rocsparse[rocSPARSE]
365369
rocsolver[rocSOLVER]
366-
hipblaslt[hipBLASLt]
367-
hipsparselt[hipSPARSELt]
370+
rocsparse[rocSPARSE]
371+
migraphx[MIGraphX]
368372
miopen[MIOpen]
369373
rccl[RCCL]
374+
rocprim[rocPRIM]
375+
hipblas[hipBLAS]
376+
hipcub[hipCUB]
370377
hipfft[hipFFT]
371378
hiprand[hipRAND]
372-
hipblas[hipBLAS]
373-
hipsparse[hipSPARSE]
374379
hipsolver[hipSOLVER]
380+
hipsparse[hipSPARSE]
375381
end
376382
377-
rocfft --> rocmstack
378-
rocrand --> rocmstack
379-
rocblas --> rocmstack
380-
rocsparse --> rocmstack
381-
rocsolver --> rocmstack
382-
rocsolver --> rocblas
383383
hipblaslt --> rocmstack
384384
hipblaslt --> roctracer
385385
hipsparselt --> rocmstack
386386
hipsparselt --> roctracer
387387
hipsparselt --> hipsparse
388+
rocblas --> rocmstack
389+
rocfft --> rocmstack
390+
rocrand --> rocmstack
391+
rocsolver --> rocmstack
392+
rocsolver --> rocblas
393+
rocsparse --> rocmstack
388394
395+
migraphx --> rocmstack
396+
migraphx --> miopen
397+
migraphx --> rocblas
389398
miopen --> rocmstack
390-
miopen --> rocblas
391399
miopen --> hipblaslt
400+
miopen --> rocblas
392401
miopen --> hipblas
393402
394403
rccl --> rocmstack
404+
rocprim --> rocmstack
395405
396-
hipfft --> rocfft
397-
hiprand --> rocrand
398406
hipblas --> rocblas
399-
hipblas --> rocsparse
400407
hipblas --> rocsolver
401-
hipsparse --> rocsparse
408+
hipblas --> rocsparse
409+
hipcub --> rocprim
410+
hipfft --> rocfft
411+
hiprand --> rocrand
402412
hipsolver --> rocsolver
413+
hipsparse --> rocsparse
403414
```
404415

416+
## Case Studies
417+
418+
For some commonly used software we examined the ROCm dependencies to make our overview more comprehensive.
419+
Specifically, we examined which ROCm libraries and frameworks they depended on.
420+
Below you can find a list of dependencies for each case study.
421+
422+
### PyTorch ROCm Dependencies
423+
424+
* HIP
425+
* hipBLAS
426+
* hipBLASLt
427+
* hipFFT
428+
* hipRAND
429+
* hipSOLVER
430+
* hipSPARSE
431+
* hipSPARSELt
432+
* MIOpen
433+
* ROCm-LLVM
434+
* ROCm-SMI
435+
* rocBLAS
436+
* rocFFT
437+
* rocRAND
438+
* rocSOLVER
439+
* roc-tracer
440+
441+
### TensorFlow ROCm Dependencies
442+
443+
* HIP
444+
* hipBLAS
445+
* hipCUB
446+
* hipFFT
447+
* hipRAND
448+
* hipSOLVER
449+
* hipSPARSE
450+
* MIGraphX
451+
* MIOpen
452+
* RCCL
453+
* rocBLAS
454+
* rocFFT
455+
* rocPRIM
456+
* rocRAND
457+
* rocSOLVER
458+
459+
### GROMACS ROCm Dependencies
460+
461+
* HIP
462+
* ROCm-LLVM
463+
* rocFFT
464+
405465
## Compatibility Policies {: #Compatibility-Policies }
406466

407467
[Source](https://rocm.docs.amd.com/en/latest/compatibility/compatibility-matrix.html)
@@ -579,3 +639,8 @@ Vega refers to AMD's GPU architecture that was one of the first to fully support
579639
* fixed dev tools dependencies graph
580640
* sorted libraries
581641
* removed the big dependencies graph in favour of dedicated ones
642+
643+
## v.6.4.1-20250620
644+
645+
* added the MIGraphX, rocPRIM, hipCUB libraries
646+
* added case studies for PyTorch, TensorFlow, GROMACS

0 commit comments

Comments
 (0)