Bartłomiej Kocot 240330481c Add blockwise gemm to ck wrapper (#1139) ...

* Add blockwise gemm to ck wrapper

* Add blockwise gemm traits

* Disable test_gemm for non xdl devices

* Fixes

* Add c layout descritpions

[ROCm/composable_kernel commit: f3b6c23ac5]

2024-01-31 21:24:40 +01:00

..

batched_gemm

Fix MNKPadding in gridwise_gemm_xdlops_v2r3 (#981)

2023-10-10 21:48:07 +02:00

batched_gemm_gemm

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

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

batched_gemm_multi_d

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

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

batched_gemm_reduce

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

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

batched_gemm_softmax_gemm

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

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

batched_gemm_softmax_gemm_permute

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

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

batchnorm

Resolve some data type issues and cmake policy. (#940)

2023-09-26 08:39:11 -07:00

block_swizzle_test

initial stream-k implementation with example (#699)

2023-07-26 14:18:15 -05:00

block_to_ctile_map

update copyright headers (#726)

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

contraction

Add support for mixed precision in contraction scale and bilinear (#973)

2023-11-02 14:26:33 -07:00

conv_tensor_rearrange

Fixing most of the cppcheck errors. (#1142)

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

conv_util

update copyright headers (#726)

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

convnd_bwd_data

Enable gfx941 and gfx942 architectures. (#752)

2023-06-15 08:20:59 -07:00

convnd_fwd

Enable gfx941 and gfx942 architectures. (#752)

2023-06-15 08:20:59 -07:00

data_type

Fix cmake dtype check (#989)

2023-10-21 22:19:43 +02:00

elementwise_normalization

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

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

gemm

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

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

gemm_layernorm

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

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

gemm_reduce

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

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

gemm_split_k

Add an option to change the number of warm-up cycles and iterations. (#1124)

2024-01-09 09:43:08 -08:00

grouped_convnd_bwd_data

Add grouped conv bwd data wmma (#950)

2023-09-28 23:10:18 +02:00

grouped_convnd_bwd_weight

Add grouped conv bwd weight wmma (#985)

2023-10-17 10:32:26 +02:00

grouped_convnd_fwd

Introduce multiABD api and deprecate multiD (#1035)

2023-11-14 17:00:40 +01:00

grouped_gemm

Add an option to change the number of warm-up cycles and iterations. (#1124)

2024-01-09 09:43:08 -08:00

magic_number_division

update copyright headers (#726)

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

normalization_bwd_data

layernorm and groupnorm backward data (#1083)

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

normalization_bwd_gamma_beta

layernorm & groupnorm bwd gamma beta (#1133)

2024-01-25 19:53:15 +08:00

normalization_fwd

layernorm and groupnorm backward data (#1083)

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

permute_scale

Hip tensor permute unit test (#1068)

2023-12-18 21:35:00 -06:00

pool

MaxPool & AvgPool bwd instances, test, ckProfiler, client example (#861)

2023-08-31 21:01:50 +08:00

reduce

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

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

reference_conv_fwd

update copyright headers (#726)

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

softmax

[HotFix] add config and version files to pass on build info (#856)

2023-08-23 11:36:17 -07:00

space_filling_curve

update copyright headers (#726)

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

transpose

Transpose profiler fix (#1114)

2024-01-04 10:33:19 -06:00

wmma_op

update copyright headers (#726)

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

wrapper

Add blockwise gemm to ck wrapper (#1139)

2024-01-31 21:24:40 +01:00

CMakeLists.txt

layernorm & groupnorm bwd gamma beta (#1133)

2024-01-25 19:53:15 +08:00