torch.xpu =================================== .. automodule:: torch.xpu .. currentmodule:: torch.xpu .. autosummary:: :toctree: generated :nosignatures: StreamContext current_device current_stream device device_count device_of empty_cache get_device_capability get_device_name get_device_properties init is_available is_initialized set_device set_stream stream synchronize Random Number Generator ------------------------- .. autosummary:: :toctree: generated :nosignatures: get_rng_state get_rng_state_all initial_seed manual_seed manual_seed_all seed seed_all set_rng_state set_rng_state_all Streams and events ------------------ .. autosummary:: :toctree: generated :nosignatures: Event Stream .. This module needs to be documented. Adding here in the meantime .. for tracking purposes .. py:module:: torch.xpu.random .. py:module:: torch.xpu.streams