torch_sys_cuda/
lib.rs

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/// A companion to the `torch-sys` crate that provides bindings for
10/// CUDA-specific functionality from libtorch. This crate is separated out to
11/// make it easier for clients who want to avoid compiling CUDA code.
12///
13/// The same safety logic described in the `torch-sys` crate applies here.
14mod bridge;
15pub mod cuda;
16pub mod nccl;