Lightning-AI/lightning-thunder

Make `numel` a method of TensorProxy not a property attribute

Open

#925 aperta il 5 ago 2024

Vedi su GitHub
 (2 commenti) (0 reazioni) (0 assegnatari)Python (114 fork)github user discovery
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.

Guida contributor