Skip to content

Fix FROM_BEST_AVAILABLE matching with hl.specialize() after #1883#1940

Open
fulvius31 wants to merge 1 commit intopytorch:mainfrom
fulvius31:fix-from-best-hspecialize
Open

Fix FROM_BEST_AVAILABLE matching with hl.specialize() after #1883#1940
fulvius31 wants to merge 1 commit intopytorch:mainfrom
fulvius31:fix-from-best-hspecialize

Conversation

@fulvius31
Copy link
Copy Markdown
Collaborator

Summary

  • Use _base_specialization_key() in FROM_BEST_AVAILABLE cache lookups to match the base key format already used by cache storage
  • Previously, the query used specialization_key() (which includes hl.specialize() extras), while the cache stores _base_specialization_key() — the mismatch meant FROM_BEST_AVAILABLE silently returned zero matches for any kernel using hl.specialize()

Test

  • Added test_find_similar_matches_with_specialize_extras

@fulvius31 fulvius31 requested review from hinriksnaer and jansel April 3, 2026 14:24
@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Apr 3, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant