Bartłomiej Kocot fd2feb1419 Support large tensors in grouped conv fwd (#1332) ...

* Support large tensors in grouped conv fwd

* Multi ABD fixes

* Fix calculate element space size

2024-06-27 10:06:40 -07:00

..

batched_gemm

Split the instances by architecture. (#1223)

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

batched_gemm_gemm

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -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

Split the instances by architecture. (#1223)

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

batched_gemm_softmax_gemm

Split the instances by architecture. (#1223)

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

batched_gemm_softmax_gemm_permute

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -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

Extend support for contraction 6D (#1207)

2024-04-09 23:46:21 +02: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

Split the instances by architecture. (#1223)

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

convnd_fwd

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -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_add

Split the instances by architecture. (#1223)

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

gemm_layernorm

Split the instances by architecture. (#1223)

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

gemm_reduce

Split the instances by architecture. (#1223)

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

gemm_split_k

Split the instances by architecture. (#1223)

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

gemm_universal

[GEMM] UniversalGemm update (#1262)

2024-04-26 12:56:07 -05:00

grouped_convnd_bwd_data

Split the instances by architecture. (#1223)

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

grouped_convnd_bwd_weight

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

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

grouped_convnd_fwd

Support large tensors in grouped conv fwd (#1332)

2024-06-27 10:06:40 -07:00

grouped_gemm

Add unit tests for grouped gemm two stage (#1256)

2024-05-15 10:03:39 +02:00

magic_number_division

update copyright headers (#726)

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

normalization_bwd_data

Grouped GEMM Multiple D tile loop. (#1247)

2024-04-25 15:12:53 -05:00

normalization_bwd_gamma_beta

Grouped GEMM Multiple D tile loop. (#1247)

2024-04-25 15:12:53 -05:00

normalization_fwd

layernorm and groupnorm backward data (#1083)

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

permute_scale

Split the instances by architecture. (#1223)

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

pool

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

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

position_embedding

[CK_TILE] support alibi (#1269)

2024-05-07 22:32:54 +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

Split the instances by architecture. (#1223)

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

wmma_op

update copyright headers (#726)

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

wrapper

Split the instances by architecture. (#1223)

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

CMakeLists.txt

Select appropriate GPU targets for instances, tests, and examples. (#1304)

2024-05-22 11:45:27 -07:00