index.rst 629 Bytes
Newer Older
Przemek Tredak's avatar
Przemek Tredak committed
1
..
2
    Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
Przemek Tredak's avatar
Przemek Tredak committed
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

    See LICENSE for license information.

C/C++ API
=========

.. Caution:: This feature is not officially supported yet and may change without notice.

The C/C++ API allows you to access the custom kernels defined in `libtransformer_engine.so` library
directly from C/C++, without Python.

.. toctree::
   :caption: Headers

   activation.h <activation>
   cast.h <cast>
   gemm.h <gemm>
cyanguwa's avatar
cyanguwa committed
20
   fused_attn.h <fused_attn>
Przemek Tredak's avatar
Przemek Tredak committed
21
   layer_norm.h <layer_norm>
22
   softmax.h <softmax>
Przemek Tredak's avatar
Przemek Tredak committed
23
24
   transformer_engine.h <transformer_engine>
   transpose.h <transpose>