torch.accelerator

This package introduces support for the current accelerator in python.

device_count	Return the number of current accelerator available.
is_available	Check if the current accelerator is available at runtime: it was build, all the required drivers are available and at least one device is visible.
current_accelerator	Return the device of the accelerator available at compilation time.
set_device_index	Set the current device index to a given device.
set_device_idx	Set the current device index to a given device.
current_device_index	Return the index of a currently selected device for the current accelerator.
current_device_idx	Return the index of a currently selected device for the current accelerator.
set_stream	Set the current stream to a given stream.
current_stream	Return the currently selected stream for a given device.
synchronize	Wait for all kernels in all streams on the given device to complete.