1 | /* |
---|---|
2 | * Copyright (c) Meta Platforms, Inc. and affiliates. |
3 | * All rights reserved. |
4 | * |
5 | * This source code is licensed under the BSD-style license found in the |
6 | * LICENSE file in the root directory of this source tree. |
7 | */ |
8 | |
9 | #pragma once |
10 | |
11 | #include <tensorpipe/config_cuda.h> |
12 | |
13 | // High-level API |
14 | |
15 | #include <tensorpipe/common/cuda_buffer.h> |
16 | |
17 | // Channels |
18 | |
19 | #include <tensorpipe/channel/cuda_basic/factory.h> |
20 | #include <tensorpipe/channel/cuda_xth/factory.h> |
21 | |
22 | #if TENSORPIPE_HAS_CUDA_GDR_CHANNEL |
23 | #include <tensorpipe/channel/cuda_gdr/factory.h> |
24 | #endif // TENSORPIPE_HAS_CUDA_GDR_CHANNEL |
25 | |
26 | #if TENSORPIPE_HAS_CUDA_IPC_CHANNEL |
27 | #include <tensorpipe/channel/cuda_ipc/factory.h> |
28 | #endif // TENSORPIPE_HAS_CUDA_IPC_CHANNEL |
29 |