composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 09:16:52 +00:00

Files

Adam Osewski 3da5c19e62 Softmax client example (#396 )

* Update Softmax device operation interface.

* Update ckProfiler.

* Update Softmax UT.

* Update example.

* Client example.

* Clang format

Co-authored-by: Adam Osewski <aosewski@amd.com>

2022-09-06 12:22:48 -05:00

01_gemm

minor fix in gemm client example (#328 )

2022-07-13 10:54:38 -05:00

02_gemm_add_add_fastgelu

Clean up conv example, Instances, profiler and test (#324 )

2022-07-29 18:19:25 -05:00

03_gemm_layernorm

Batchnorm-forward and Batchnorm-infer Implemented using generic kernels (#320 )

2022-08-15 10:11:02 -05:00

04_contraction

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

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

05_layernorm

layernorm external api (#379 )

2022-08-24 18:43:43 -05:00

06_softmax

Softmax client example (#396 )

2022-09-06 12:22:48 -05:00

CMakeLists.txt

Softmax client example (#396 )

2022-09-06 12:22:48 -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