Cecilia/feature/nki-all-operators - #259
Open
Cecilia123li wants to merge 116 commits into
Open
Cecilia123li wants to merge 116 commits into
Cecilia123li wants to merge 116 commits into
Conversation
added 30 commits
June 8, 2026 19:29
…m allowed partitions and the free dimensions (PMAX, F) instead of (N, 1). Also removed unnecessary imports.
… up to 64k elements for SBUF. No pre-defined attribute for the maximum sbuf free dimensions at this moment so I am using a hard-coded value.
This was referenced Aug 29, 2026
Collaborator
…rflow fix Bowen's autotune commit built on top of the old host-side _choose_oh_block() helper (guessed SBUF_BUDGET_BYTES=100KB) that was removed after it overflowed real SBUF. Keep the in-kernel auto-sizing (nl.tile_size.sbuf_fmax_bytes, resolved inside the trace) as the default path via an oh_block=0 sentinel, and let NkiAutotuner override it with candidate block sizes -- each one is validated against the same real SBUF budget inside the kernel, so a candidate that doesn't fit raises and is skipped by tune_or_cached rather than trusted. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_013qAybi1Ym2hrjarDPdyGrs
…ector-add # Conflicts: # core/engine.py # core/nki_timer.py # scripts/run_bench.py
bowencui123
added a commit
that referenced
this pull request
Sep 18, 2026
…ches
The open pull requests were written against the old source layout. This
moves any file still under a pre-refactor source directory onto the
tilebench package and rewrites the imports inside it, so each branch is a
merge plus one command:
git merge -c merge.directoryRenames=false origin/main
python scripts/migrate_pr_paths.py
Disabling directory-rename detection is load-bearing. With it on, git
guesses a destination for a file added inside a renamed directory and picks
the wrong one for the NKI operator PRs, offering
tilebench/benchmarks/llm_generated/<op>/<model>/high/final/impl_nki.py
instead of tilebench/benchmarks/operators/<op>/impl_nki.py.
Verified on #294 (single operator), #303 (cross-operator import) and #259
(45 operators at once). Temporary: delete it once those branches land.
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01EdNBRsQvET5PUJ3Nw1hVCA
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
All 45 operators + changes in core files to support nki trainium