pytorch/torch/_library
PyTorch MergeBot e9bd74d763 Revert "[export] don't decompose custom triton op when exporting (#142426)"
This reverts commit 10b9c5944e.

Reverted https://github.com/pytorch/pytorch/pull/142426 on behalf of https://github.com/huydhn due to This fails one internal MTIA test, checking with the author that we need to revert and reland this ([comment](https://github.com/pytorch/pytorch/pull/142426#issuecomment-2555793496))
2024-12-19 21:21:38 +00:00
..
__init__.py
autograd.py
custom_ops.py Improve torch.library.opcheck and register_autograd docs (#141883) 2024-12-03 16:28:56 +00:00
fake_class_registry.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
fake_impl.py
infer_schema.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
simple_registry.py
triton.py Revert "[export] don't decompose custom triton op when exporting (#142426)" 2024-12-19 21:21:38 +00:00
utils.py