ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-05 22:22:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
bd09b5c5389c9f7f51459e5a38d3d7253b0c8dc0
composable_kernel/include/ck/tensor_operation/gpu/grid
History
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
..
batchnorm_multiblock
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gemm_layernorm
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
normalization
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
block_to_ctile_map.hpp
Grouped Gemm with Fixed K and N with SplitK (#818)
2023-08-31 09:22:12 -05:00
gridwise_2d_multiple_reduction_multiblock.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_2d_multiple_reduction_threadwise.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_2d_reduction_multiblock.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_2d_reduction_threadwise.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_batched_gemm_gemm_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_batched_gemm_multiple_d_gemm_multiple_d_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_batched_gemm_multiple_d_softmax_gemm_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_batchnorm_backward_blockwise_welford.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_batchnorm_forward_blockwise_welford.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_elementwise_1d.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_elementwise_2d.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_elementwise_layernorm_welford_variance.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_gemm_bias_add_reduce_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_dl_multiple_d.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_gemm_dl_v1r3.hpp
Redesign the DPP8 GEMM kernel to use warp-wise component (#863)
2023-09-06 11:44:09 -05:00
gridwise_gemm_dpp.hpp
Enable DPP8 GEMM on Navi3 (#892)
2023-09-08 11:14:57 -05:00
gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
Contraction multi abd (#957)
2023-10-02 09:18:36 -05:00
gridwise_gemm_multiple_d_multiple_r_xdl_cshuffle.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_multiple_d_wmma_cshuffle.hpp
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gridwise_gemm_multiple_d_xdl_cshuffle.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_multiple_d_xdl_splitk_cshuffle.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_pipeline_selector.hpp
Split GEMM instance library & enable pipeline v2 optimization (#783)
2023-07-06 10:59:35 -05:00
gridwise_gemm_pipeline_v1.hpp
Redesign the DPP8 GEMM kernel to use warp-wise component (#863)
2023-09-06 11:44:09 -05:00
gridwise_gemm_pipeline_v2.hpp
Fixed Gemmv2r3 kpad (#938)
2023-09-26 18:40:00 -05:00
gridwise_gemm_pipeline_v3.hpp
initial stream-k implementation with example (#699)
2023-07-26 14:18:15 -05:00
gridwise_gemm_reduce_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle_v2.hpp
Fp16/fp8 mixed-precision Gemm with multiply+add fusion (#865)
2023-08-28 16:27:32 -05:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_waveletmodel.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_gemm_wmma.hpp
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gridwise_gemm_xdl_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdl_layernorm_cshuffle_v1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdl_waveletmodel_cshuffle.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_bwd_weight.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_skip_b_lds_v1.hpp
Enable gfx941 and gfx942 architectures. (#752)
2023-06-15 08:20:59 -07:00
gridwise_gemm_xdlops_streamk.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_v2r3.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_v2r4.hpp
Enable gfx941 and gfx942 architectures. (#752)
2023-06-15 08:20:59 -07:00
gridwise_gemm_xdlops_v2r4r2.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_v3r1.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_v3r2.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_gemm_xdlops_v3r3.hpp
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
gridwise_permute.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_put_element_1d.hpp
Average pool backward deviceOP and example (#797)
2023-08-10 12:04:35 +08:00
gridwise_set_buffer_value.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_set_multiple_buffer_value.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_softmax.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_sparse_embeddings_forward_layernorm.hpp
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gridwise_tensor_rearrange.hpp
Add column to image kernel (#930)
2023-09-27 17:19:06 +02:00