composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00

Files

rocking5566 fe367bc917 Refine layernorm naming and test code (#497 )

* Sync the naming

* Sync the test of layernorm with groupnorm

* Sync the naming

* Minor change for comment and log

* [What] Add saveMean and SaveInvVariance in the interface.
[Why] These can optimize the backward

[ROCm/composable_kernel commit: d4d1147f0a]

2022-11-02 16:57:28 -06: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

Refactor device op implementations into impl subdirectory. (#420 )

2022-10-13 09:05:08 -05:00

04_contraction

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

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

05_layernorm

Refine layernorm naming and test code (#497 )

2022-11-02 16:57:28 -06: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

08_fused_attention

Fused attention client example (#494 )

2022-10-27 15:04:31 -06:00

09_grouped_conv2d_bwd_data

Add client example of grouped conv2d backward data (data type: fp16) (#481 )

2022-11-02 16:54:41 -06:00

09_quantization

Conv perlayer int8 quantization (#471 )

2022-11-02 13:56:26 -06: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