Skip to content

[HINT] Add Triton v3.3.x hint manager#348

Merged
zhzhcookie merged 7 commits intotriton_v3.3.xfrom
triton_v3.3.x_hint_manager
Mar 11, 2026
Merged

[HINT] Add Triton v3.3.x hint manager#348
zhzhcookie merged 7 commits intotriton_v3.3.xfrom
triton_v3.3.x_hint_manager

Conversation

@starrryz
Copy link
Copy Markdown
Collaborator

@starrryz starrryz commented Feb 6, 2026

Summary
This PR migrates the HintManager architecture to the v3.3.x branch. It decouples backend logic from the core compiler, enabling cleaner support for AIPU and Ascend backends without polluting the main codebase.

Key Changes
Architecture: Introduced HintManager and BaseHintHandler to manage backend-specific logic via a standardized hook mechanism.

Decoupling: Refactored CodeGenerator.py and jit.py by replacing hardcoded if backend == ... checks with generic hint_trigger calls.

AIPU Support: Added AipuHintHandler to handle backend-specific logic (e.g., tl.load hints), separating invocation logic from core infrastructure changes.

Optimization: Implemented lazy loading for handlers to avoid circular imports and standardized source code comment parsing.

@starrryz starrryz requested a review from zhzhcookie as a code owner February 6, 2026 08:15
sgjzfzzf
sgjzfzzf previously approved these changes Feb 6, 2026
Copy link
Copy Markdown
Collaborator

@sgjzfzzf sgjzfzzf left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Codes look good to me. Please let others also take a look.

@zhzhcookie zhzhcookie changed the title [FEAT] Triton v3.3.x hint manager [HINT] Add Triton v3.3.x hint manager Mar 10, 2026
@starrryz starrryz force-pushed the triton_v3.3.x_hint_manager branch from feb9b3a to 70877e3 Compare March 10, 2026 06:32
@starrryz starrryz force-pushed the triton_v3.3.x_hint_manager branch from 70877e3 to 1ffece8 Compare March 11, 2026 02:34
Copy link
Copy Markdown
Collaborator

@zhzhcookie zhzhcookie left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@zhzhcookie zhzhcookie merged commit 10f748f into triton_v3.3.x Mar 11, 2026
6 checks passed
@zhzhcookie zhzhcookie deleted the triton_v3.3.x_hint_manager branch March 11, 2026 04:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants