pytorch/torch/_library
Yidi Wu 567552b98b fix typo in doc and import for torch._library.triton (#144882)
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
2025-01-17 17:32:12 +00:00
..
__init__.py fix typo in doc and import for torch._library.triton (#144882) 2025-01-17 17:32:12 +00:00
autograd.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
custom_ops.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +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 [real_tensor_prop] Infer Fake kernels during real tensor prop (#139213) 2024-10-30 21:29:33 +00:00
infer_schema.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
simple_registry.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
triton.py fix typo in doc and import for torch._library.triton (#144882) 2025-01-17 17:32:12 +00:00
utils.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00