ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 10:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
225c98244db0d2efef3602236437d8013efa4611
composable_kernel/include/ck/tensor_operation/gpu/device
History
zjing14 71758e1559 Grouped GEMM for fp16 (#126) 
* init of grouped_gemm

* 2 gemm test

* perf test

* clean

* wrap desc into a struct

* test cast static_arr to pointer

* add ptr to GemmDesc

* add grouped gemm profiler

* fixed mem issue with unique_ptr

* clean

* clean

* finished ckprofiler

* Update README.md

* readme

* fixed readme

* add example

* improve code

* fixed comments: reserve, seperate ptr and gemm_shapes

* merge group and non-group

* fixed comments: replace push_back with emplace_back to avoid copy constructor

* fixed comments: unified blk2ctile; add test

* ci fix

* fixed ci

* fixed ci

* fixed ci

[ROCm/composable_kernel commit: 716f1c7fb1]
2022-03-22 18:18:18 -05:00
..
conv_utils.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
convolution_backward_data_specialization.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
convolution_forward_specialization.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06: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_xdl.hpp
refactored deviceBatchedGemm; removed GridwiseBatchedGemm; added fp32 and int8 to profiler (#120)
2022-03-21 16:45:14 -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
Fix conv2d bwd data bug when filter is 1x1 and stride = 2 (#132)
2022-03-21 10:53:23 -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
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06: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
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06: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_fwd_xdl_nhwc_kyxc_nhwk.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06: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_xdl_c_shuffle_bias_2d.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06: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_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
Grouped GEMM for fp16 (#126)
2022-03-22 18:18:18 -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
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
reduction_operator_mapping.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
tensor_layout.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00