Skip to content

Commit 892ffde

Browse files
committed
Update documentation for AMD GPU architecture
1 parent 9d4a67e commit 892ffde

File tree

1 file changed

+51
-51
lines changed

1 file changed

+51
-51
lines changed

docs/source/keywords.rst

+51-51
Original file line numberDiff line numberDiff line change
@@ -494,77 +494,77 @@ Kokkos will attempt to autodetect the architecture flag at configuration time.
494494
AMD GPUs
495495
~~~~~~~~
496496

497+
The Kokkos naming convention is to aggregate AMD\_ and the architecture flag.
498+
499+
``Kokkos_ARCH_AMD_<ARCHITECTURE_FLAG>``
500+
501+
If the HIP backend is enabled and no AMD GPU architecture is specified,
502+
Kokkos will attempt to autodetect the architecture flag at configuration time.
503+
497504
.. list-table::
498-
:widths: 25 65 10
505+
:widths: 30 15 25 30
499506
:header-rows: 1
500507
:align: left
501508

502509
* - **AMD GPUs**
503-
- Description/info
504-
- Default
510+
- Architecture flags
511+
- Models
512+
- Notes
513+
514+
* * ``Kokkos_ARCH_AMD_GFX942``
515+
* GFX942
516+
* MI300A, MI300X
517+
* (since Kokkos 4.2)
518+
519+
* * ``Kokkos_ARCH_AMD_GFX940``
520+
* GFX940
521+
* MI300A (A0 part)
522+
* (since Kokkos 4.2.1)
505523

506524
* * ``Kokkos_ARCH_AMD_GFX90A``
507-
* Optimize for AMD GPU MI200 series GFX90A :sup:`since Kokkos 4.2`
508-
* ``OFF``
525+
* GFX90A
526+
* MI200 series
527+
* (since Kokkos 4.2)
509528

510529
* * ``Kokkos_ARCH_AMD_GFX908``
511-
* Optimize for AMD GPU MI100 GFX908 :sup:`since Kokkos 4.2`
512-
* ``OFF``
530+
* GFX90A
531+
* MI100
532+
* (since Kokkos 4.2)
513533

514534
* * ``Kokkos_ARCH_AMD_GFX906``
515-
* Optimize for AMD GPU MI50/MI60 GFX906 :sup:`since Kokkos 4.2`
516-
* ``OFF``
535+
* GFX906
536+
* MI50, MI60
537+
* (since Kokkos 4.2)
517538

518539
* * ``Kokkos_ARCH_AMD_GFX1100``
519-
* Optimize for AMD GPU 7900xt GFX1100 :sup:`since Kokkos 4.2`
520-
* ``OFF``
540+
* GFX1100
541+
* 7900xt
542+
* (since Kokkos 4.2)
521543

522544
* * ``Kokkos_ARCH_AMD_GFX1030``
523-
* Optimize for AMD GPU V620/W6800 GFX1030 :sup:`since Kokkos 4.2`
524-
* ``OFF``
525-
526-
* * ``Kokkos_ARCH_VEGA900``
527-
* Optimize for AMD GPU MI25 GFX900 :sup:`removed in 4.0`
528-
* ``OFF``
529-
530-
* * ``Kokkos_ARCH_VEGA906``
531-
* Optimize for AMD GPU MI50/MI60 GFX906 (Prefer ``Kokkos_ARCH_AMD_GFX906``)
532-
* ``OFF``
533-
534-
* * ``Kokkos_ARCH_VEGA908``
535-
* Optimize for AMD GPU MI100 GFX908 (Prefer ``Kokkos_ARCH_AMD_GFX908``)
536-
* ``OFF``
545+
* GFX1030
546+
* V620, W6800
547+
* (since Kokkos 4.2)
537548

538549
* * ``Kokkos_ARCH_VEGA90A``
539-
* Optimize for AMD GPU MI200 series GFX90A (Prefer ``Kokkos_ARCH_AMD_GFX90A``)
540-
* ``OFF``
550+
* GFX90A
551+
* MI200 series
552+
* Prefer ``Kokkos_ARCH_AMD_GFX90A``
541553

554+
* * ``Kokkos_ARCH_VEGA908``
555+
* GFX908
556+
* MI100
557+
* Prefer ``Kokkos_ARCH_AMD_GFX908``
542558

543-
.. list-table::
544-
:widths: 65 35
545-
:header-rows: 1
546-
:align: left
559+
* * ``Kokkos_ARCH_VEGA906``
560+
* GFX906
561+
* MI50, MI60
562+
* Prefer ``Kokkos_ARCH_AMD_GFX906``
547563

548-
* - AMD GPU
549-
- Kokkos ARCH
550-
551-
* * ``7900xt``
552-
* AMD_GFX1100
553-
554-
* * ``MI50/MI60``
555-
* AMD_GFX906
556-
557-
* * ``MI100``
558-
* AMD_GFX908
559-
560-
* * ``MI200`` series: ``MI210``, ``MI250``, ``MI250X``
561-
* AMD_GFX90A
562-
563-
* * ``V620``
564-
* AMD_GFX1030
565-
566-
* * ``W6800``
567-
* AMD_GFX1030
564+
* * ``Kokkos_ARCH_VEGA900``
565+
* GFX900
566+
* MI25
567+
* removed in 4.0
568568

569569
Intel GPUs
570570
~~~~~~~~~~

0 commit comments

Comments
 (0)