"Fossies" - the Fresh Open Source Software Archive

Member "pytorch-1.8.2/docs/source/cuda.rst" (23 Jul 2021, 1752 Bytes) of package /linux/misc/pytorch-1.8.2.tar.gz:


As a special service "Fossies" has tried to format the requested source page into HTML format (assuming markdown format). Alternatively you can here view or download the uninterpreted source code file. A member file download can also be achieved by clicking within a package contents listing on the according byte size field. See also the last Fossies "Diffs" side-by-side code changes report for "cuda.rst": 1.11.0_vs_1.12.0.

torch.cuda

torch.cuda

torch.cuda

Random Number Generator

get_rng_state

get_rng_state_all

set_rng_state

set_rng_state_all

manual_seed

manual_seed_all

seed

seed_all

initial_seed

Communication collectives

torch.cuda.comm.broadcast

torch.cuda.comm.broadcast_coalesced

torch.cuda.comm.reduce_add

torch.cuda.comm.scatter

torch.cuda.comm.gather

Streams and events

Stream

Event

Memory management

empty_cache

list_gpu_processes

memory_stats

memory_summary

memory_snapshot

memory_allocated

max_memory_allocated

reset_max_memory_allocated

memory_reserved

max_memory_reserved

set_per_process_memory_fraction

memory_cached

max_memory_cached

reset_max_memory_cached

NVIDIA Tools Extension (NVTX)

torch.cuda.nvtx.mark

torch.cuda.nvtx.range_push

torch.cuda.nvtx.range_pop