mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 09:08:35 +00:00

Files

Mateusz Ozga 35d8d114a5 Apply universal gemm to bwd_weight_cshuffle operator (#1873 )

* Universal gemm - initial commit

* Review part 1

* Fix tests

* Remove instances

* Remove comp instances

[ROCm/composable_kernel commit: c287418dcc]

2025-02-18 10:10:22 +01:00

01_gemm

Fix test_gemm_universal test_grouped_convnd_fwd and example_gemm_xdl_streamk on gfx950 (#1891 )

2025-02-17 11:06:45 -08:00

02_gemm_bilinear

…

03_gemm_bias_relu

…

04_gemm_add_add_fastgelu

Merge from internal (#1857 )

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

09_convnd_fwd

…

10_convnd_fwd_multiple_d_multiple_reduce

Fix example_convnd_fwd_max_xdl_int8 failures on MI300 (#1666 )

2024-11-14 08:40:50 -08:00

12_reduce

…

13_pool2d_fwd

…

14_gemm_quantization

…

15_grouped_gemm

OCP FP8 support for gfx12. (#1710 )

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

16_gemm_multi_d_multi_reduces

Fix example_convnd_fwd_max_xdl_int8 failures on MI300 (#1666 )

2024-11-14 08:40:50 -08:00

17_convnd_bwd_data

…

18_batched_gemm_reduce

Merge from internal (#1857 )

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

19_binary_elementwise

…

20_grouped_conv_bwd_weight

Apply universal gemm to bwd_weight_cshuffle operator (#1873 )

2025-02-18 10:10:22 +01:00

21_gemm_layernorm

OCP FP8 support for gfx12. (#1710 )

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

22_cgemm

…

23_softmax

…

24_batched_gemm

Added Int4 mixed batch gemm support (#1839 )

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

25_gemm_bias_e_permute

…

26_contraction

…

27_layernorm2d_fwd

…

28_grouped_gemm_bias_e_permute

…

29_batched_gemm_bias_e_permute

…

30_grouped_conv_fwd_multiple_d

Merge from internal (#1857 )

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

31_batched_gemm_gemm

Merge from internal (#1857 )

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

32_batched_gemm_scale_softmax_gemm

OCP FP8 support for gfx12. (#1710 )

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

33_multiple_reduce

…

34_batchnorm

Refactor elementwise kernels (#1222 )

2024-04-19 13:31:17 +02:00

35_splitK_gemm

OCP FP8 support for gfx12. (#1710 )

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

36_sparse_embedding

…

37_batched_gemm_add_add_relu_gemm_add

OCP FP8 support for gfx12. (#1710 )

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

38_grouped_conv_bwd_data_multiple_d

OCP FP8 support for gfx12. (#1710 )

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

39_permute

…

40_conv2d_fwd_quantization

…

41_grouped_conv_conv_fwd

Merge from internal (#1857 )

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

42_groupnorm_fwd

…

43_splitk_gemm_bias_e_permute

…

44_elementwise_permute

Prevent instantiation of undefined FP8 operators. (#1639 )

2024-11-05 13:58:29 -08:00

45_elementwise_normalization

…

46_gemm_add_multiply

…

47_gemm_bias_softmax_gemm_permute

OCP FP8 support for gfx12. (#1710 )

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

48_pool3d_fwd

…

49_maxpool2d_bwd

…

50_put_element

…

51_avgpool3d_bwd

…

52_im2col_col2im

…

53_layernorm2d_bwd

…

54_groupnorm_bwd

…

59_grouped_gemm_multi_ABD

OCP FP8 support for gfx12. (#1710 )

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

60_gemm_multi_ABD

…

61_contraction_multi_ABD

…

62_convnd_activ

Merge from internal (#1857 )

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

63_layernorm4d_fwd

…

64_fpAintB_gemm

…

65_gemm_multiply_multiply

OCP FP8 support for gfx12. (#1710 )

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

66_complex_contraction_bilinear

update layernorm (#1570 )

2024-10-22 09:26:18 +08:00

67_gemm_microscaling

Add a host mx gemm reference kernel (#1864 )

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

ck_tile

Fix test_gemm_universal test_grouped_convnd_fwd and example_gemm_xdl_streamk on gfx950 (#1891 )

2025-02-17 11:06:45 -08:00

CMakeLists.txt

Merge from internal (#1857 )

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

README.md

Add basic documentation structure (#1715 )

2024-12-04 00:46:47 +01:00

README.md

Back to the main page

Composable Kernel examples