composable_kernel

batchnorm_multiblock

…

gemm_layernorm

…

normalization

…

block_to_ctile_map.hpp

Codegen hipRTC compilation (#1579 )

2025-01-31 09:48:39 -08:00

gridwise_2d_multiple_reduction_multiblock.hpp

Fix for beta!=0 in reduce (#1440 )

2024-08-06 09:10:39 -07:00

gridwise_2d_multiple_reduction_threadwise.hpp

…

gridwise_2d_reduction_multiblock.hpp

Fix for beta!=0 in reduce (#1440 )

2024-08-06 09:10:39 -07:00

gridwise_2d_reduction_threadwise_multi_d.hpp

Universal gemm splitk using reduce (with multi-d) (#1341 )

2024-07-19 22:01:22 +08:00

gridwise_2d_reduction_threadwise.hpp

…

gridwise_batched_gemm_gemm_xdl_cshuffle_v1.hpp

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

gridwise_batched_gemm_multiple_d_gemm_multiple_d_xdl_cshuffle_v1.hpp

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

gridwise_batched_gemm_multiple_d_softmax_gemm_xdl_cshuffle_v1.hpp

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp

Merging the gfx12 code into public repo. (#1362 )

2024-06-27 00:33:34 -07:00

gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

gridwise_batchnorm_backward_blockwise_welford.hpp

…

gridwise_batchnorm_forward_blockwise_welford.hpp

…

gridwise_elementwise_1d_scale.hpp

…

gridwise_elementwise_2d.hpp

Add support for NGCHW in grouped conv bwd wei (#1491 )

2024-09-03 10:52:03 +02:00

gridwise_elementwise_layernorm_welford_variance.hpp

…

gridwise_fpAintB_gemm_wmma.hpp

Merging the gfx12 code into public repo. (#1362 )

2024-06-27 00:33:34 -07:00

gridwise_gemm_bias_add_reduce_xdl_cshuffle_v1.hpp

…

gridwise_gemm_dl_multiple_d.hpp

Add Grouped Conv Fwd Large Tensor kernel (#1432 )

2024-08-06 10:06:10 +02:00

gridwise_gemm_dl_v1r3.hpp

Add Grouped Conv Fwd Large Tensor kernel (#1432 )

2024-08-06 10:06:10 +02:00

gridwise_gemm_dpp.hpp

…

gridwise_gemm_multiple_abd_xdl_cshuffle.hpp

Codegen hipRTC compilation (#1579 )

2025-01-31 09:48:39 -08:00

gridwise_gemm_multiple_d_multiple_r_xdl_cshuffle.hpp

…

gridwise_gemm_multiple_d_wmma_cshuffle.hpp

Statically Cast Pointer Offset (#1631 )

2024-11-05 09:59:08 -08:00

gridwise_gemm_multiple_d_xdl_cshuffle_lds_direct_load.hpp

Change flag to CK_GFX90A_DENORM_WORKAROUND (#1817 )

2025-01-28 09:58:39 -05:00

gridwise_gemm_multiple_d_xdl_cshuffle.hpp

Codegen hipRTC compilation (#1579 )

2025-01-31 09:48:39 -08:00

gridwise_gemm_multiple_d_xdl_splitk_cshuffle.hpp

…

gridwise_gemm_pipeline_selector.hpp

Codegen hipRTC compilation (#1579 )

2025-01-31 09:48:39 -08:00

gridwise_gemm_pipeline_v1.hpp

…

gridwise_gemm_pipeline_v2.hpp

…

gridwise_gemm_pipeline_v3.hpp

…

gridwise_gemm_pipeline_v4_direct_load.hpp

…

gridwise_gemm_reduce_xdl_cshuffle_v1.hpp

…

gridwise_gemm_split_k_multiple_d_xdl_cshuffle_v2.hpp

…

gridwise_gemm_split_k_multiple_d_xdl_cshuffle.hpp

…

gridwise_gemm_waveletmodel.hpp

…

gridwise_gemm_wmma.hpp

Merging the gfx12 code into public repo. (#1362 )

2024-06-27 00:33:34 -07:00

gridwise_gemm_xdl_cshuffle_bwd_weight_v3.hpp

Optimize grouped conv bwd weight for small M and N (#1303 )

2024-05-22 21:01:01 +02:00

gridwise_gemm_xdl_cshuffle_streamk_v3.hpp

universal streamk fp8 changes (#1665 )

2024-11-21 08:21:37 -08:00

gridwise_gemm_xdl_cshuffle_v1.hpp

…

gridwise_gemm_xdl_cshuffle_v2.hpp

…

gridwise_gemm_xdl_cshuffle_v3_b_preshuffle.hpp

move b thread dequant copy to blockwise.

2025-02-13 14:34:59 +08:00

gridwise_gemm_xdl_cshuffle_v3_b_scale.hpp

Added Int4 mixed batch gemm support (#1839 )

2025-02-10 11:17:02 +08:00

gridwise_gemm_xdl_cshuffle_v3_multi_abd.hpp

Optimize grouped conv bwd weight for small M and N (#1303 )

2024-05-22 21:01:01 +02:00

gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp

OCP FP8 support for gfx12. (#1710 )

2024-12-03 08:42:55 -08:00

gridwise_gemm_xdl_cshuffle_v3_multi_d.hpp

Add SplitK support into Batched GEMM V3 (#1729 )

2024-12-13 21:08:35 +01:00

gridwise_gemm_xdl_cshuffle_v3.hpp

Jing's contribution: prototype of mixed precision gemm FP16/BF16xint4 GEMM (#1762 )

2025-01-02 11:48:06 +08:00

gridwise_gemm_xdl_layernorm_cshuffle_v1.hpp

…

gridwise_gemm_xdl_waveletmodel_cshuffle.hpp

…

gridwise_gemm_xdlops_bwd_weight.hpp

Change flag to CK_GFX90A_DENORM_WORKAROUND (#1817 )

2025-01-28 09:58:39 -05:00

gridwise_gemm_xdlops_skip_b_lds_v1.hpp

…

gridwise_gemm_xdlops_splitk_lds_direct_load.hpp

aggregate device macros in ck_tile config header (#1297 )

2024-05-20 08:34:45 -07:00

gridwise_gemm_xdlops_streamk.hpp

…

gridwise_gemm_xdlops_v2r3.hpp

Change flag to CK_GFX90A_DENORM_WORKAROUND (#1817 )

2025-01-28 09:58:39 -05:00

gridwise_gemm_xdlops_v2r4.hpp

…

gridwise_gemm_xdlops_v2r4r2.hpp

…

gridwise_gemm_xdlops_v3r1.hpp

…

gridwise_gemm_xdlops_v3r2.hpp

…

gridwise_gemm_xdlops_v3r3.hpp

…

gridwise_permute.hpp

…

gridwise_put_element_1d.hpp

…

gridwise_set_buffer_value.hpp

…

gridwise_set_multiple_buffer_value.hpp

…

gridwise_softmax.hpp

…

gridwise_sparse_embeddings_forward_layernorm.hpp

…

gridwise_tensor_rearrange.hpp

Statically Cast Pointer Offset (#1631 )

2024-11-05 09:59:08 -08:00