composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00

Files

Jianfeng Yan 87522d2ac9 Batched gemm and reduction (#156 )

* adding batched_gemm_and_reduction

* batched_gemm_reduce works with bactch_count=1

* fix a bug in grid_size; batched_gemm_reduce works for batch_count > 1

* adding profiler for batched_gemm_fp16

* fixed a bug in declaration of d1 and d0; both example and profiler work

* clang-format

* cleanup

* batched_gemm_reduce: add test

* minor change

* fixed some typo in function names

[ROCm/composable_kernel commit: 34c661e71c]

2022-03-30 11:21:18 -05:00

profile_batched_gemm_impl.hpp

Batched gemm and reduction (#156 )

2022-03-30 11:21:18 -05:00

profile_batched_gemm_reduce_impl.hpp

Batched gemm and reduction (#156 )

2022-03-30 11:21:18 -05:00

profile_conv_bwd_data_impl.hpp

Unified implementation of 1d/2d/3d conv bwd-data. fp32/fp16/bfp16/int8 (#134 )

2022-03-29 10:52:25 -05:00

profile_conv_fwd_bias_relu_add_impl.hpp

GEMM+Bias+ReLU+Add (#76 )