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

    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

15
   transformer_engine.h <transformer_engine>
Przemek Tredak's avatar
Przemek Tredak committed
16
   activation.h <activation>
17
   cast_transpose_noop.h <cast_transpose_noop>
Przemek Tredak's avatar
Przemek Tredak committed
18
   cast.h <cast>
19
   cudnn.h <cudnn>
cyanguwa's avatar
cyanguwa committed
20
   fused_attn.h <fused_attn>
21
22
   fused_rope.h <fused_rope>
   gemm.h <gemm>
23
   multi_tensor.h <multi_tensor>
24
25
26
27
   normalization.h <normalization>
   padding.h <padding>
   permutation.h <permutation>
   recipe.h <recipe>
28
   softmax.h <softmax>
29
   swizzle.h <swizzle>
Przemek Tredak's avatar
Przemek Tredak committed
30
   transpose.h <transpose>