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