mirror of
https://github.com/nomic-ai/kompute.git
synced 2026-05-11 17:09:59 +00:00
127 lines
3.9 KiB
ReStructuredText
127 lines
3.9 KiB
ReStructuredText
|
|
Class Documentation and C++ Reference
|
|
========
|
|
|
|
This section provides a breakdown of the cpp classes and what each of their functions provide. It is partially generated and augomented from the Doxygen autodoc content. You can also go directly to the `raw doxygen docs <../doxygen/annotated.html>`_.
|
|
|
|
Below is a diagram that provides insights on the relationship between Vulkan Kompute objects and Vulkan resources, which primarily encompass ownership of either CPU and/or GPU memory.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture.jpg
|
|
:width: 100%
|
|
|
|
Manager
|
|
-------
|
|
|
|
The Kompute Manager provides a high level interface to simplify interaction with underlying kp::Sequences of kp::Operations.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-manager.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::Manager
|
|
:members:
|
|
|
|
Sequence
|
|
-------
|
|
|
|
The Kompute Sequence consists of batches of kp::Operations, which are executed on a respective GPU queue. The execution of sequences can be synchronous or asynchronous, and it can be coordinated through its respective vk::Fence.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-sequence.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::Sequence
|
|
:members:
|
|
|
|
Tensor
|
|
-------
|
|
|
|
The kp::Tensor is the atomic unit in Kompute, and it is used primarily for handling Host and GPU Device data.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-tensor.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::Tensor
|
|
:members:
|
|
|
|
Algorithm
|
|
-------
|
|
|
|
The kp::Algorithm consists primarily of the components required for shader code execution, including the relevant vk::DescriptorSet relatedresources as well as vk::Pipeline and all the relevant Vulkan resources as outlined in the architectural diagram.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-algorithm.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::Algorithm
|
|
:members:
|
|
|
|
OpBase
|
|
-------
|
|
|
|
The kp::OpBase provides a top level class for an operation in Kompute, which is the step that is executed on a GPU submission. The Kompute operations can consist of one or more kp::Tensor.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-operations.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::OpBase
|
|
:members:
|
|
|
|
OpAlgoBase
|
|
-------
|
|
|
|
The vk::OpAlgoBase extends the vk::OpBase class, and provides the base for shader-based operations. Besides of consisting of one or more vk::Tensor as per the vk::OpBase, it also contains a unique vk::Algorithm.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-opmult.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::OpAlgoBase
|
|
:members:
|
|
|
|
OpMult
|
|
-------
|
|
|
|
The kp::OpMult operation is a sample implementation of the kp::OpAlgoBase class. This class shows how it is possible to create a custom vk::OpAlgoBase that can compile as part of the binary. The kp::OpMult operation uses the shader-to-cpp-header-file script to convert the script into cpp header files.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-opmult.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::OpMult
|
|
:members:
|
|
|
|
OpTensorCreate
|
|
-------
|
|
|
|
The kp::OpTensorCreate is a tensor only operations which initialises a kp::Tensor by creating the respective vk::Buffer and vk::Memory, as well as transferring the local data into the GPU.
|
|
|
|
.. image:: ../images/kompute-vulkan-architecture-opcreatetensor.jpg
|
|
:width: 100%
|
|
|
|
.. doxygenclass:: kp::OpTensorCreate
|
|
:members:
|
|
|
|
OpTensorCopy
|
|
-------
|
|
|
|
The kp::OpTensorCopy is a tensor only operation that copies the GPU memory buffer data from one kp::Tensor to one or more subsequent tensors.
|
|
|
|
.. doxygenclass:: kp::OpTensorCopy
|
|
:members:
|
|
|
|
OpTensorSyncLocal
|
|
-------
|
|
|
|
The kp::OpTensorSyncLocal is a tensor only operation that maps the data from the GPU device memory into the local host vector.
|
|
|
|
.. doxygenclass:: kp::OpTensorSyncLocal
|
|
:members:
|
|
|
|
OpTensorSyncDevice
|
|
-------
|
|
|
|
The kp::OpTensorSyncDevice is a tensor only operation that maps the data from the local host vector into the GPU device memory.
|
|
|
|
.. doxygenclass:: kp::OpTensorSyncDevice
|
|
:members:
|
|
|
|
|
|
|
|
|