Zum Hauptinhalt springen
••••••••••••••••••••
Intelligence View
⚡ tsecurity.de Intelligence

ciflow/torchtitan/185992: Implement get_stride for PermuteView (#185992)

Summary: What it does: Adds a get_stride() method to the PermuteView class in torch/_inductor/ir.py. The method permutes the strides of the underlying data in…

Beitrag
0
Seite
0
↗ Quelle (github.com)
Social ReaktionenReagiere als Erste:r — dein Feedback zählt!

Summary:


What it does:


Adds a get_stride() method to the PermuteView class in torch/_inductor/ir.py. The method permutes the strides of the underlying data in the same order as the dimensions — mirroring the existing get_size() pattern


Why:


Previously, PermuteView didn't implement get_stride(), which caused a NotImplementedError when downstream operations (like aten.slice.Tensor) needed stride info from a PermuteView wrapping a ComputedBuffer. This was hit during AOTI lowering of models that do compute → permute → slice (e.g., pad → permute → slice in the GR model).


Test Plan:



  • internal lowering passed with this patch

  • CI for added unit test


Reviewed By: silverlakeli, ColinPeppler


Differential Revision: D107267848

Ähnliche Beiträge
🔍 Verwandte News

Auch interessante Nachrichten ciflow/torchtitan/185992: Implement get_stride for PermuteView (#185992)

Thematisch verwandte Begriffe: ciflowtorchtitan185992, Implement, getstride, PermuteView · 6 Treffer

Laden...

Videos werden geladen ...

Laden...

Beiträge werden geladen ...

Laden...

Videos werden geladen ...

Laden...

Beiträge werden geladen ...

Laden...

Videos werden geladen ...

Laden...

Beiträge werden geladen ...

Laden...

Videos werden geladen ...

Laden...

Beiträge werden geladen ...

Laden...

Videos werden geladen ...

💬 Kommentare werden geladen…
Zum Aktualisieren ziehen
tsecurity.de Icon
Offline-Lesen, Eilmeldungen & 0ms Ladezeit

Installiere tsecurity.de direkt auf deinen Home-Bildschirm für das ultimative Vollbild-Magazinerlebnis ohne Browser-Leisten.

Nächster Beitrag