Lightning-AI/lightning-thunder
Vedi su GitHubMake `numel` a method of TensorProxy not a property attribute
Open
#925 aperta il 5 ago 2024
good first issue
Metriche repository
- Star
- (1460 star)
- Metriche merge PR
- (Metriche PR in attesa)
Descrizione
🐛 Bug
PyTorch uses tensor.numel() while Thunder's TensorProxy and in the codebase it's used as a property tensor.numel. The property attribute makes more sense, but it conflicts with setting PyTorch as the default language context (https://github.com/Lightning-AI/lightning-thunder/pull/924).
Related PR: https://github.com/Lightning-AI/lightning-thunder/pull/278
We should decide what to do with numel and how to avoid the discrepancy with PyTorch.