index.rst 562 Bytes
Newer Older
Przemek Tredak's avatar
Przemek Tredak committed
1
..
2
    Copyright (c) 2022-2024, 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

    See LICENSE for license information.

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

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
18
   fused_attn.h <fused_attn>
Przemek Tredak's avatar
Przemek Tredak committed
19
   layer_norm.h <layer_norm>
20
   rmsnorm.h <rmsnorm>
21
   softmax.h <softmax>
Przemek Tredak's avatar
Przemek Tredak committed
22
23
   transformer_engine.h <transformer_engine>
   transpose.h <transpose>