ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
9d58c42103613ad09725e44b29316eba99ddc916
composable_kernel/example/61_contraction_multi_ABD
History
zjing14 9d58c42103 Contraction multi abd (#957) 
* add gridwise_multi_abd

* move element_op into RunRead

* merge element_wise op with data read

* add multiABD example

* allow packed elementwise_op

* changed example

* clean

* clean

* add is_detected

* fix

* minor fix

* add scaleAdd_vec4 example

* init commit for contraction_multi_ABD

* add examples

* add examples of multiA and broadcast

* update example

* fixed comments

* Update cmake-ck-dev.sh

* Update cmake-ck-dev.sh

* Add comments into the example

---------

Co-authored-by: Jing Zhang <jizha@amd.com>
2023-10-02 09:18:36 -05:00
..
CMakeLists.txt
Contraction multi abd (#957)
2023-10-02 09:18:36 -05:00
contraction_multi_ABD_xdl_fp16.cpp
Contraction multi abd (#957)
2023-10-02 09:18:36 -05:00