what-is-ck.rst 1.3 KB
Newer Older
randyh62's avatar
randyh62 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
.. meta::
  :description: Composable Kernel documentation and API reference library
  :keywords: composable kernel, CK, ROCm, API, documentation

.. _what-is-ck:

********************************************************************
What is the Composable Kernel library
********************************************************************


Methodology
===========

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

CK utilizes two concepts to achieve performance portability and code maintainability:

* A tile-based programming model
* Algorithm complexity reduction for complex ML operators using an innovative technique called
  "Tensor Coordinate Transformation".

randyh62's avatar
randyh62 committed
23
.. image:: ../data/ck_component.png
randyh62's avatar
randyh62 committed
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
   :alt: CK Components


Code Structure
==============

The CK library is structured into 4 layers:

* "Templated Tile Operators" layer
* "Templated Kernel and Invoker" layer
* "Instantiated Kernel and Invoker" layer
* "Client API" layer

It also includes a simple wrapper component used to perform tensor transform operations more easily and with fewer lines of code.

randyh62's avatar
randyh62 committed
39
.. image:: ../data/ck_layer.png
randyh62's avatar
randyh62 committed
40
41
   :alt: CK Layers