Yanbo Liang ec91b7720f [Custom Ops] Add a new API to allow users to register an autocast for the custom op (#145588) ...

Fixes #137033

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145588
Approved by: https://github.com/zou3519

2025-01-27 19:22:43 +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

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102)

2025-01-18 20:47:12 +00:00

custom_ops.py

[Custom Ops] Add a new API to allow users to register an autocast for the custom op (#145588)

2025-01-27 19:22:43 +00:00

fake_class_registry.py

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102)

2025-01-18 20:47:12 +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

PEP585 update - torch/_dynamo (#145105)

2025-01-18 20:47:11 +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

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102)

2025-01-18 20:47:12 +00:00

utils.py

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102)

2025-01-18 20:47:12 +00:00