Add support for tagging parameters in export

pytorch / torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

BSD 3-Clause "New" or "Revised" License

1.01k stars 124 forks source link

Closed voznesenskym closed 1 year ago

voznesenskym commented 1 year ago

Today, we only have support for tagging / arbitrary __dict__ code in export for inputs. We want to extend this to params.