mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00

PyTorch MergeBot 6341311333 Revert "Add unified memory APIs for torch.accelerator (#152932 )"

This reverts commit 2ad5c25cfc.

Reverted https://github.com/pytorch/pytorch/pull/152932 on behalf of https://github.com/ZainRizvi due to Very sorry but this is still breaking internally. @albanD would you be able to help get this past the finish line? D78496124 has more details on the failure and the workaround might be to do something like what's in D78684669. To validate the fixes internally, you can follow the instructions here to ghimport the changes: https://fburl.com/fixing-ghfirst-reverts ([comment](https://github.com/pytorch/pytorch/pull/138222#issuecomment-3100195370))

2025-07-22 01:01:41 +00:00

422 B

Raw Blame History

torch.accelerator

.. automodule:: torch.accelerator

.. currentmodule:: torch.accelerator

.. autosummary::
    :toctree: generated
    :nosignatures:

    device_count
    is_available
    current_accelerator
    set_device_index
    set_device_idx
    current_device_index
    current_device_idx
    set_stream
    current_stream
    synchronize
    device_index

422 B Raw Blame History

torch.accelerator

422 B

Raw Blame History