CK Docker Hub

2. CK Docker Hub#

2.1. Why do I need this?#

To make our lives easier and bring Composable Kernel dependencies together, we recommend using docker images that can be found on Docker Hub.

2.2. So what is Composable Kernel?#

Composable Kernel (CK) library aims to provide a programming model for writing performance critical kernels for machine learning workloads across multiple architectures including GPUs, CPUs, etc, through general purpose kernel languages, like HIP C++.

To get the CK library:

git clone https://github.com/ROCmSoftwarePlatform/composable_kernel.git

run a docker container:

docker run                                                            \
-it                                                                   \
--privileged                                                          \
--group-add sudo                                                      \
-w /root/workspace                                                    \
-v ${PATH_TO_LOCAL_WORKSPACE}:/root/workspace                         \
rocm/composable_kernel:ck_ub20.04_rocm5.6                             \
/bin/bash

and build the CK:

mkdir build && cd build
# Need to specify target ID, example below is for gfx908 and gfx90a
cmake                                                                                             \
-D CMAKE_PREFIX_PATH=/opt/rocm                                                                    \
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc                                                         \
-D CMAKE_CXX_FLAGS="-O3"                                                                          \
-D CMAKE_BUILD_TYPE=Release                                                                       \
-D GPU_TARGETS="gfx908;gfx90a"                                                                    \
..

and:

make -j examples tests

To run all the test cases including tests and examples run:

make test

We can also run specific examples or tests like:

./bin/example_gemm_xdl_fp16
./bin/test_gemm_fp16

For more details visit CK github repository, CK examples, even more CK examples.

2.3. And what is inside?#

The docker images have everything you need for running CK including:

2.4. Which image is right for me?#

Let’s take a look at the image naming, for example ck_ub20.04_rocm5.6. The image specs are:

  • ck - made for running Composable Kernel;

  • ub20.04 - based on Ubuntu 20.04;

  • rocm5.6 - ROCm platform version 5.6.

So just pick the right image for your project dependencies and you’re all set.

2.5. DIY starts here#

If you need to customize a docker image or just can’t stop tinkering, feel free to adjust the Dockerfile for your needs.

2.6. License#

CK is released under the MIT license.

Contents