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
ada72bead282f2be63ebbb22c484ade37293355c
composable_kernel/include/ck/tensor_operation/gpu/grid
History
rocking5566 ada72bead2 Gemm multiple d multiple r (#335) 
* Imitate XXX_gemm_multiple_d, add XXX_gemm_multiple_d_multiple_r for gemm + reduction

* Implement run of kernel

* Add example

* Fix parameter of typo

* Rewrite the reduceMax example

* Rewrite the reduceMean + reduceMeanSquare example

* Refine naming

* Refine folder name

* refine naming

* Rewrite the gemm + bias + relu + add + layernorm example

* Rewrite the gemm + layernorm example

* clang-format

* Fix bug if sync lds

* Fix compile error

[ROCm/composable_kernel commit: 6c3c06bf1f]
2022-08-13 01:07:12 -05:00
..
block_to_ctile_map.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_2d_reduction_multiblock.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_2d_reduction_threadwise.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_5ary_Elementwise_1d.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
Fused attention (#345)
2022-08-13 00:16:14 -05:00
gridwise_binary_elementwise_1d.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_contraction_dlops_v1r2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_bias_add_reduce_xdl_cshuffle_v1.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
gridwise_gemm_dl_v1r3.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_dlops_v1r2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_dlops_v2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_dlops_v3.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_multiple_d_multiple_r_xdl_cshuffle.hpp
Gemm multiple d multiple r (#335)
2022-08-13 01:07:12 -05:00
gridwise_gemm_multiple_d_xdl_cshuffle.hpp
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
gridwise_gemm_pipeline_v1.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_pipeline_v2.hpp
GEMM pipeline v2 (#317)
2022-07-08 15:55:14 -05:00
gridwise_gemm_reduce_xdl_cshuffle_v1.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
gridwise_gemm_xdl_cshuffle_v1.hpp
GEMM pipeline v2 (#317)
2022-07-08 15:55:14 -05:00
gridwise_gemm_xdl_layernorm_cshuffle_v1.hpp
Single-kernel GEMM + layernorm (#263)
2022-07-01 01:38:00 -05:00
gridwise_gemm_xdlops_bwd_weight.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v2r3.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v2r4.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v2r4r2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v3r1.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v3r2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_gemm_xdlops_v3r3.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_layernorm.hpp
Standalone layernorm (#315)
2022-07-13 11:16:14 -05:00
gridwise_set_buffer_value.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gridwise_softmax.hpp
fix standalone softmax race condition around blockwise reduction (#323)
2022-07-14 22:52:45 -05:00
gridwise_unary_elementwise_1d.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00