flash-attention/csrc/composable_kernel/docs/index.rst

.. meta::
  :description: Composable Kernel documentation and API reference library
  :keywords: composable kernel, CK, ROCm, API, documentation

.. _composable-kernel:

********************************************************************
Composable Kernel User Guide
********************************************************************

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++. This document contains instructions for installing, using, and contributing to the Composable Kernel project. To learn more see :ref:`what-is-ck`.

The CK documentation is structured as follows:

.. grid:: 2
  :gutter: 3

  .. grid-item-card:: Installation

    * :ref:`docker-hub`

  .. grid-item-card:: Conceptual

    * :ref:`what-is-ck`

  .. grid-item-card:: API reference

    * :ref:`supported-primitives`
    * :ref:`api-reference`
    * :ref:`wrapper`

  .. grid-item-card:: Tutorial

    * :ref:`hello-world`

To contribute to the documentation refer to `Contributing to ROCm  <https://rocm.docs.amd.com/en/latest/contribute/contributing.html>`_.

You can find licensing information on the `Licensing <https://rocm.docs.amd.com/en/latest/about/license.html>`_ page.