ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 05:31:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,370 Commits 734 Branches 37 Tags
f161b5b738781c71bd5f2c191561b81f679ba9ed
Commit Graph

3 Commits

Author SHA1 Message Date
Rostyslav Geyyer
aa932445ea Add a gpu gemm reference kernel (#1528) 
* Add a gpu gemm reference kernel

* Switch to gpu reference in gemm examples

* Remove redundant arguments

* Update all related examples

* Update more examples

* Try less threads per block

* Try even less threads per block

* Add support for all matrix layouts

* Increase block size

* Clean up

* Remove hardcoded strides

* Clean up

* Try a column-major case

* Revert back to row-major

* Run both CPU and GPU veriffication

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
 2024-10-08 11:05:28 -05:00
Rostyslav Geyyer
363feb482d Refactor tolerances for correctness check in gemm op (#1188) 
* Refactor tolerances for correctness check

* Update tolerances

* Update host-side gemm

* Update reference gemm call
 2024-03-08 12:05:05 -08:00
Rostyslav Geyyer
bd09b5c538 Add fp8 @ bf8 gemm support and example (#933) 
* Add f8 bf8 gemm example

* Add element-wise ops

* Add intrinsics

* Update reference calculation

* Add an additional type option for xdlops gemm

* Fix build process

* Add bf8 to buffer addressing

* Update blockwise op, split typeA and typeB

* Update for compatibility

* Uppdate naming to f8->fp8

* Update naming

* Format
 2023-10-02 16:39:03 -05:00