pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

George White 940b60db97 Use the device interface for detecting Triton availability (#139171 )

This allows for each device type to check current devices for Triton compatibility and ensure their Triton backend is present.

This PR replaces the `has_triton()` global method which was previously used for this task, and moves the initial check for each Inductor backend on to their associated `BaseScheduler` subclass. This means that other backends, such as Halide, can also implement their own availability checks.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139171
Approved by: https://github.com/jansel

2025-03-11 03:56:11 +00:00

__init__.py

PEP585: More UP006 fixes (#146392 )

2025-02-20 06:18:13 +00:00

_semi_structured_conversions.py

Remove unused Python variables in torch/[b-z]* (#136963 )

2024-10-19 16:45:22 +00:00

_semi_structured_ops.py

[sparse] add extra options to _cslt_spare_mm (#137427 )

2024-11-27 05:32:45 +00:00

_triton_ops_meta.py

PEP585 update - torch/nn torch/optim torch/package torch/profiler torch/serialization torch/sparse torch/xpu (#145175 )