Brian K. Ryu 147f5673d0 New RMS Norm example with unit tests (#2917) ...

* Add rmsnorm example

* Address reviewer comments. (1) use the cute.runtime definition directly. (2) use the nvvm_wrapper's warp reduce directly

* Separate out reduce.py

* Change copyright notice years

2026-01-13 09:05:31 +08:00

..

blockwise_gemm

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

mamba2_ssd

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

mixed_input_fmha

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

tutorial_gemm

add pytest support for tutorial gemm (#2826)

2025-12-05 08:45:01 -05:00

dense_blockscaled_gemm_persistent_prefetch.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_blockscaled_gemm_persistent.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_gemm_alpha_beta_persistent.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_gemm_persistent_prefetch.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_gemm_persistent.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_gemm_software_pipeline.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

dense_gemm.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

fmha_bwd.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

fmha.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

grouped_blockscaled_gemm.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

grouped_gemm.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

grouped_mixed_input_gemm.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

mixed_input_gemm.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

mla.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

programmatic_dependent_launch.py

v4.3.5 update. (#2934)

2026-01-08 15:02:56 -05:00

reduce.py

New RMS Norm example with unit tests (#2917)

2026-01-13 09:05:31 +08:00

rmsnorm.py

New RMS Norm example with unit tests (#2917)

2026-01-13 09:05:31 +08:00