frozenleaves/pytorch

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

Files

History

Vinayak Pawar 9ad5e8edb1 Improve typing of ONNX decorators with ParamSpec (#162332 )

## Summary
This PR improves typing in ONNX-related modules by replacing TypeVar bound to Callable[..., Any] with ParamSpec to preserve parameter types and avoid type erasure in decorator functions.

## Changes
- `torch/onnx/_internal/exporter/_flags.py`: Replace TCallable TypeVar with ParamSpec
- `torch/onnx/ops/_impl.py`: Replace _T TypeVar with ParamSpec for _onnx_op decorator
- `torch/onnx/_internal/exporter/_torchlib/_torchlib_registry.py`: Replace _T TypeVar with ParamSpec

## Motivation
The previous implementation used TypeVar bound to Callable which erased parameter type information to Any. ParamSpec preserves the exact parameter types and return types, providing better type safety and IDE support.

## Testing
- Verified all changes compile and import correctly
- Created comprehensive test suite to validate ParamSpec functionality
- No linting errors introduced
- Maintains backward compatibility

Fixes #142306
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162332
Approved by: https://github.com/Skylion007

2025-09-07 18:06:03 +00:00

..

Improve typing of ONNX decorators with ParamSpec (#162332 )

2025-09-07 18:06:03 +00:00

Improve typing of ONNX decorators with ParamSpec (#162332 )

2025-09-07 18:06:03 +00:00

__init__.py

[ONNX] Default to dynamo export (#159646 )

2025-09-02 22:45:55 +00:00

_constants.py

[ONNX] Set default opset to 20 (#158802 )

2025-07-22 19:55:05 +00:00

_flags.py

[ONNX] Hide draft export under a flag (#162225 )

2025-09-05 19:54:50 +00:00

errors.py

[ONNX] Clean up the diagnostics module (#149864 )

2025-03-26 05:58:32 +00:00

operators.py

remove allow-untyped-defs from torch/onnx/operators.py (#144133 )

2025-01-03 20:07:56 +00:00

README.md

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_helper.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset7.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset8.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset9.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset10.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset11.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset12.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset13.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset14.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset15.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset16.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset17.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset18.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset19.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

symbolic_opset20.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

utils.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

verification.py

[ONNX] Refactor torchscript based exporter (#161323 )

2025-09-02 16:10:30 +00:00

README.md

torch.onnx

Torch->ONNX converter / exporter.

User-facing docs: https://pytorch.org/docs/main/onnx.html
Developer docs: https://github.com/pytorch/pytorch/wiki/PyTorch-ONNX-exporter