composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 08:15:04 +00:00

Files

Chao Liu 4fe9c393b8 N-D Tensor Contraction example, instance, and client example (#270 )

* 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

2022-07-07 14:31:11 -05:00

01_gemm

Improve external interface for GEMM and GEMM+add+add+fastgelu (#311 )

2022-06-30 22:11:00 -05:00

02_gemm_add_add_fastgelu

Improve external interface for GEMM and GEMM+add+add+fastgelu (#311 )

2022-06-30 22:11:00 -05:00

03_gemm_layernorm

Improve external interface for GEMM and GEMM+add+add+fastgelu (#311 )

2022-06-30 22:11:00 -05:00

04_contraction

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

2022-07-07 14:31:11 -05:00

CMakeLists.txt

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

2022-07-07 14:31:11 -05:00

README.md

Improve external interface for GEMM and GEMM+add+add+fastgelu (#311 )

2022-06-30 22:11:00 -05:00

README.md

Client application links to CK library, and therefore CK library needs to be installed before building client applications.

Build

mkdir -p client_example/build
cd client_example/build

cmake                                                                 \
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc                             \
-D CMAKE_PREFIX_PATH="/opt/rocm;${PATH_TO_CK_INSTALL_DIRECTORY}"      \
..

Build client example

 make -j