Enrico Degregori ce99cab605 Wmma support for gemm_ab_scale (#3314) ...

* Support gemm_ab_scale:

 - Add tests
 - Integrate scaling implementation in multiple D
 - Generalize existing b_scale for ab_scale
 - Add instances
 - Generalize implementation for ScaleBlockM, ScaleBlockN, ScaleBlockK
 - Add support for all layouts supported by xdl
 - Fix splitk xdl

* Fix copyright

* Wmma support for gemm_blockscale_wp (#3315)

* Support for  preshuffle with ab scale

 - add support for b preshuffle in GridwiseGemm_wmma_cshuffle_v3_ab_scale
 - add support for AScaleLayout amnd BScaleLayout (can be different
   from ALayout and BLayout, respectively)
 - add Run method in v1 pipeline to support preshuffle + scaling
 - add support for preshuffle gemms in common invoker
 - Add splitk support

* Fix copyright header

2025-12-11 09:06:20 +01:00

..

CMakeLists.txt

Wmma support for gemm_ab_scale (#3314)

2025-12-11 09:06:20 +01:00

test_gemm_blockscale_wp_fp8.cpp

Wmma support for gemm_ab_scale (#3314)

2025-12-11 09:06:20 +01:00

test_gemm_common.hpp

chore(copyright): update copyright header for test directory (#3265)

2025-11-22 19:38:27 -05:00