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
efd1d25733fb22f4900698d86dd88599e7864102
composable_kernel/example/36_sparse_embedding
History
carlushuang efd1d25733 embedding fuse layernorm (#405) 
* add gridwise/device sparse embedding

* update code

* update code

* remove useless makefile

* code fix

* workable

* work properly

* emb add

* add more instance

* format

* remove useless code

* fix format

* fix clang-tidy

* clean

* fix a compile error

Co-authored-by: Chao Liu <chao.liu2@amd.com>
Co-authored-by: Chao Liu <lc.roy86@gmail.com>
2022-09-09 10:41:15 -05:00
..
CMakeLists.txt
embedding fuse layernorm (#405)
2022-09-09 10:41:15 -05:00
sparse_embedding3_forward_layernorm.cpp
embedding fuse layernorm (#405)
2022-09-09 10:41:15 -05:00