NVIDIA/Fuser

Add more features to the pre-segmentation pass to move take_along_axis ahead of certain unary ops

Open

#4.491 geöffnet am 21. Mai 2025

Auf GitHub ansehen
 (0 Kommentare) (0 Reaktionen) (1 zugewiesene Person)C++ (80 Forks)github user discovery
New Opsgood first issue

Repository-Metriken

Stars
 (392 Stars)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

The PR add a pre-segmentation pass that moved a take_along_axis operator ahead of certain unary pointwise operators such as castOp and the squeeze op.

The implementation had certain limitations. We should extend the current implementation to remove these.

  • We only consider fusions with a single take_along_axis op. We should extend our implementation to support the case where there are multiple take_along_axis ops.
  • We move a take_along_axis ahead of a squeeze op. Currently we do so only if a single dimension is being squeezed. We should remove this limitation. This will mean that we have to add multiple unsqueeze ops to the indexTv of the take_along_axis.

Contributor Guide

Research-Richtung
Studieren Sie den vorhandenen Vorsegmentierungsdurchlauf (PR #4399), um zu verstehen, wie 'take along axis' vor unäre Operationen verschoben wird. Erweitern Sie die Logik, um mehrere 'take along axis' Operationen zu unterstützen, indem Sie sie durchlaufen. Entfernen Sie für den Squeeze Fall die Einschränkung auf eine einzelne Dimension und fügen Sie für jede gequetschte Dimension mehrere 'unsqueeze' Operationen zum Indextensor hinzu.
Tech Stack
cpp
Domain
backend
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
C++
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
40