Qianfeng Zhang 240900b1c0 Add NumReduceDim template parameter to DeviceSoftmax and Softmax client API to simplify instances collecting

2023-05-30 21:17:54 +00:00

..

batched_gemm

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

batched_gemm_gemm

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

batched_gemm_reduce

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

batched_gemm_softmax_gemm

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

batched_gemm_softmax_gemm_permute

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

batchnorm

Batchnorm inference instances, external API, client examples and gtests (#531)

2023-01-25 17:09:04 -06:00

block_to_ctile_map

add license in file (#303)

2022-06-24 23:32:43 -05:00

contraction

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

conv_util

Clean up conv example, Instances, profiler and test (#324)

2022-07-29 18:19:25 -05:00

convnd_bwd_data

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

convnd_fwd

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

data_type

More int4 tests. (#374)

2022-08-25 17:20:23 -05:00

elementwise_normalization

Add instance for elementwise normlization (#573)

2023-02-09 09:37:29 -08:00

gemm

Wavelet (inter-wave consumer-producer) GEMM (#310)

2023-01-18 12:00:02 -06:00

gemm_layernorm

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

gemm_reduce

Modularize ckProfiler operations (#514)

2022-12-01 15:15:02 -06:00

gemm_split_k

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

grouped_convnd_bwd_weight

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

grouped_convnd_fwd

Modularize ckProfiler operations (#514)

2022-12-01 15:15:02 -06:00

grouped_gemm

Enable gemm_dl and other kernels on Navi3x. (#714)

2023-05-23 11:23:16 -05:00

magic_number_division

Clean up conv example, Instances, profiler and test (#324)

2022-07-29 18:19:25 -05:00

normalization

Improve normalization (#580)

2023-02-15 11:59:35 -06:00

pool_fwd

Pool3d fwd (#697)

2023-05-24 09:05:04 -05:00

reduce

Modularize ckProfiler operations (#514)

2022-12-01 15:15:02 -06:00

reference_conv_fwd

Rangify constructor of HostTensorDescriptor & Tensor<> (#445)

2022-11-11 11:36:01 -06:00

softmax

Add NumReduceDim template parameter to DeviceSoftmax and Softmax client API to simplify instances collecting

2023-05-30 21:17:54 +00:00

space_filling_curve

Input/output permutation for fused attention (#460)

2022-10-27 14:58:20 -06:00

wmma_op

[Navi3x-LWPCK-545] Block-wise GEMM + Real GEMM_WMMA_FP16 (#541)

2023-01-16 20:06:01 -06:00

CMakeLists.txt

Pool3d fwd (#697)

2023-05-24 09:05:04 -05:00