Jianfeng Yan cb97ce68d8 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

..

conv_utils.hpp

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

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

convolution_backward_data_specialization.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

convolution_forward_specialization.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

convolution_utility.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_base.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_batched_gemm_reduce_xdl_cshuffle.hpp

Batched gemm and reduction (#156)

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

device_batched_gemm_xdl.hpp

Batched gemm and reduction (#156)

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

device_conv2d_backward_weight_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk.hpp

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

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

device_conv2d_fwd_xdl_c_shuffle_bias_activation_add_nhwc_kyxc_nhwk.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv2d_fwd_xdl_c_shuffle_bias_activation_nhwc_kyxc_nhwk.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp

Unified conv3D API + support for all data types. (#133)

2022-03-23 10:23:13 -05:00

device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp

Unified conv3D API + support for all data types. (#133)

2022-03-23 10:23:13 -05:00

device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv3d_fwd_xdl_ndhwc_kzyxc_ndhwk.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

device_conv_backward_weight.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv_bwd_data.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv_fwd_bias_activation_add.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv_fwd_bias_activation.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_conv_fwd.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_convnd_bwd_data_xdl_ndhwc_kzyxc_ndhwk.hpp

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

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

device_convnd_fwd_xdl_nhwc_kyxc_nhwk.hpp

Unified conv3D API + support for all data types. (#133)

2022-03-23 10:23:13 -05:00

device_gemm_bias_activation_add.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_bias_activation.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_bias.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

device_gemm_reduce_xdl_cshuffle.hpp

Batched gemm and reduction (#156)

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

device_gemm_reduce.hpp

Batched gemm and reduction (#156)

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

device_gemm_xdl_c_shuffle_bias_2d.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

device_gemm_xdl_c_shuffle_bias_activation_add.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_xdl_c_shuffle_bias_activation.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_xdl_c_shuffle.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_xdl_cshuffle.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

device_gemm_xdl_splitk_c_shuffle.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_xdl_splitk.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm_xdl.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_gemm.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

device_grouped_gemm_xdl.hpp

Grouped GEMM for fp16 (#126)

2022-03-22 18:18:18 -05:00

device_pool2d_fwd_nhwc_nhwc.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_pool2d_fwd.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

device_reduce_blockwise_second_call.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce_blockwise.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce_common.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce_multiblock_atomic_add.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce_multiblock_partial_reduce.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce_threadwise.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

device_reduce.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00

gemm_specialization.hpp

Gemm+Reduce Fusion (#128)

2022-03-23 22:18:42 -05:00

reduction_operator_mapping.hpp

Reorganize files, Part 1 (#119)

2022-03-08 21:46:36 -06:00

tensor_layout.hpp

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

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