composable_kernel

01_gemm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

02_gemm_bilinear

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

03_gemm_bias_relu

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

04_gemm_add_add_fastgelu

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

09_convnd_fwd

Add instances for conv_scale with fp8@bf8->fp8 (#1220 )

2024-04-03 09:08:08 -05:00

10_convnd_fwd_multiple_d_multiple_reduce

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

12_reduce

Change 1d,2d,... to 1D,2D,... (#997 )

2023-10-19 16:53:18 +02:00

13_pool2d_fwd

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

14_gemm_quantization

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

15_grouped_gemm

Add Grouped Gemm Multiple D SplitK TwoStage (#1212 )

2024-04-04 11:01:33 +02:00

16_gemm_multi_d_multi_reduces

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

17_convnd_bwd_data

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

18_batched_gemm_reduce

fix merge from upstream

2024-03-26 14:09:54 +00:00

19_binary_elementwise

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

20_grouped_conv_bwd_weight

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

21_gemm_layernorm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

22_cgemm

Clean DTYPES conditions in CMake (#974 )

2023-10-18 11:14:14 -05:00

23_softmax

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

24_batched_gemm

Clean DTYPES conditions in CMake (#974 )

2023-10-18 11:14:14 -05:00

25_gemm_bias_e_permute

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

26_contraction

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

27_layernorm2d_fwd

Split the static library into several files. (#1044 )

2023-11-28 11:17:37 -08:00

28_grouped_gemm_bias_e_permute

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

29_batched_gemm_bias_e_permute

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

30_grouped_conv_fwd_multiple_d

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

31_batched_gemm_gemm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

32_batched_gemm_scale_softmax_gemm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

33_multiple_reduce

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

34_batchnorm

Batchnorm splitk single kernel (#771 )

2023-07-06 10:58:55 -05:00

35_splitK_gemm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

36_sparse_embedding

update copyright headers (#726 )

2023-05-31 18:46:57 -05:00

37_batched_gemm_add_add_relu_gemm_add

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

38_grouped_conv_bwd_data_multiple_d

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

39_permute

Clean DTYPES conditions in CMake (#974 )

2023-10-18 11:14:14 -05:00

40_conv2d_fwd_quantization

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

41_grouped_conv_conv_fwd

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

42_groupnorm_fwd

Split the static library into several files. (#1044 )

2023-11-28 11:17:37 -08:00

43_splitk_gemm_bias_e_permute

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

44_elementwise_permute

Introduce combined elementwise ops (#1217 )

2024-04-02 17:23:49 -05:00

45_elementwise_normalization

Layernorm and groupnorm support to save mean and inverse std in forward (#929 )

2023-10-19 07:36:29 +08:00

46_gemm_add_multiply

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

47_gemm_bias_softmax_gemm_permute

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

48_pool3d_fwd

Fixing most of the cppcheck errors. (#1142 )

2024-01-24 13:47:48 -08:00

49_maxpool2d_bwd

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

50_put_element

Refactoring cmake files to build data types separately. (#932 )

2023-09-20 22:15:56 -07:00

51_avgpool3d_bwd

Fixing most of the cppcheck errors. (#1142 )

2024-01-24 13:47:48 -08:00

52_im2col_col2im

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

53_layernorm2d_bwd

layernorm and groupnorm backward data (#1083 )

2023-12-19 04:23:11 +08:00

54_groupnorm_bwd

layernorm and groupnorm backward data (#1083 )

2023-12-19 04:23:11 +08:00

60_gemm_multi_ABD

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

61_contraction_multi_ABD

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

62_convnd_activ

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

63_layernorm4d_fwd

Split the static library into several files. (#1044 )

2023-11-28 11:17:37 -08:00

64_fpAintB_gemm

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00

ck_tile

Remove more not-in-use elementwise function kargs

2024-04-09 06:20:50 +00:00

CMakeLists.txt

Split the instances by architecture. (#1223 )

2024-04-02 09:42:17 -07:00