composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00

Files

rocking5566 1dcaa3991f Fix bug of layernorm ckProfiler and refine code (#448 )

* Fix bug of profiler for layernorm

* 1. Rename layernorm into normalization
2. Decouple softmax from normalization

* clang-format

[ROCm/composable_kernel commit: 1b62bfaa2a]

2022-10-12 21:06:39 -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

Fix bug of layernorm ckProfiler and refine code (#448 )

2022-10-12 21:06:39 -05:00

06_softmax

Softmax client example (#396 )

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

07_conv2d_fwd

Conv2dFwd example. (#467 )

2022-10-12 20:43:04 -05:00

CMakeLists.txt

Replace the obsolete offload-arch flags with GPU_TARGETS and fix a bug. (#437 )

2022-09-22 09:32:25 -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