mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Previously, the doc's suggested `from torch._library.triton import wrap_triton, triton_op` doesn't work because wrap_triton is not imported in torch/_library/__init__.py but `from torch.library import wrap_triton` works. This PR imports wrap_triton and fix the doc. Pull Request resolved: https://github.com/pytorch/pytorch/pull/144882 Approved by: https://github.com/zou3519 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| autograd.py | ||
| custom_ops.py | ||
| fake_class_registry.py | ||
| fake_impl.py | ||
| infer_schema.py | ||
| simple_registry.py | ||
| triton.py | ||
| utils.py | ||