File cuda.h

↰ Parent directory (torch/csrc/api/include/torch)

Definition (torch/csrc/api/include/torch/cuda.h)

• Program Listing for File cuda.h

Includes

• c10/core/Device.h (File Device.h)

• c10/macros/Export.h

• cstdint

Included By

• File Descriptors.h

• File all.h

• File data_parallel.h

Namespaces

• Namespace torch

• Namespace torch::cuda

Functions

• Function torch::cuda::device_count

• Function torch::cuda::is_available

• Function torch::cuda::manual_seed

• Function torch::cuda::manual_seed_all

• Function torch::cuda::synchronize