ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 14:59:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

N-D Tensor Contraction example, instance, and client example (#270)

Browse Source 
* adding contraction

* add contraction example

* update examle

* update example

* format

* update readme

* clean header

* clean header

* contraction with multiple D

* rename

* fix naming issue; add instances for contraction+bilinear

* change assumed virtual layout of contraction; add client example

* update example

* update

* contraction+scale

* use type_convert

* rename
This commit is contained in:
Chao Liu
2022-07-07 14:31:11 -05:00
committed by GitHub
parent 334361cbde
commit 4fe9c393b8
114 changed files with 3620 additions and 256 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
client_example/CMakeLists.txt
View File

@@ -9,3 +9,4 @@ message(STATUS "Build with HIP ${hip_VERSION}")
add_subdirectory(01_gemm)
add_subdirectory(02_gemm_add_add_fastgelu)
add_subdirectory(03_gemm_layernorm)
add_subdirectory(04_contraction)