index.rst 814 Bytes
Newer Older
Przemek Tredak's avatar
Przemek Tredak committed
1
..
2
    Copyright (c) 2022-2025, 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
20
   comm_gemm_overlap.h <comm_gemm_overlap>
   cudnn.h <cudnn>
cyanguwa's avatar
cyanguwa committed
21
   fused_attn.h <fused_attn>
22
23
   fused_rope.h <fused_rope>
   gemm.h <gemm>
24
   multi_tensor.h <multi_tensor>
25
26
27
28
   normalization.h <normalization>
   padding.h <padding>
   permutation.h <permutation>
   recipe.h <recipe>
29
   softmax.h <softmax>
30
   swizzle.h <swizzle>
Przemek Tredak's avatar
Przemek Tredak committed
31
   transpose.h <transpose>