Skip to content

Qualcomm AI Engine Direct - Change transpose output from NCH1 to NC1W before conv - #21707

Merged
psiddh merged 1 commit into
pytorch:mainfrom
CodeLinaro:dev1/chenweng/improve_prefill
Aug 13, 2026
Merged

Qualcomm AI Engine Direct - Change transpose output from NCH1 to NC1W before conv#21707
psiddh merged 1 commit into
pytorch:mainfrom
CodeLinaro:dev1/chenweng/improve_prefill

Conversation

@chenweng-quic

@chenweng-quic chenweng-quic commented Aug 10, 2026

Copy link
Copy Markdown
Collaborator

Summary

This memory layout has better performance on the HTP backend, and lead to better performance on prefill model.

Prefill performance of third partitioned graph:

Model w/o w/
llama3.2 3b instruct 55924 35774

cc @cbilgin @psiddh

@pytorch-bot

pytorch-bot Bot commented Aug 10, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21707

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 34b9aa6 with merge base fe7623e (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 10, 2026
@github-actions

Copy link
Copy Markdown

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

@chenweng-quic chenweng-quic added the module: qnn Issues related to Qualcomm's QNN delegate and code under backends/qualcomm/ label Aug 10, 2026
@psiddh

psiddh commented Aug 11, 2026

Copy link
Copy Markdown
Contributor

cc @digantdesai

@psiddh

psiddh commented Aug 11, 2026

Copy link
Copy Markdown
Contributor

@chenweng-quic Can you check the CI failures (and lint-check as well ?) Are they related ?

@chenweng-quic

Copy link
Copy Markdown
Collaborator Author

checking, will update.

@chenweng-quic
chenweng-quic force-pushed the dev1/chenweng/improve_prefill branch from afb0f01 to fc5ddcd Compare August 11, 2026 12:05
@psiddh

psiddh commented Aug 11, 2026

Copy link
Copy Markdown
Contributor

@chenweng-quic Can you check why lint-check on urls is failing ? (maybe rebase ?), once it is resolved ,we can land

@psiddh

psiddh commented Aug 11, 2026

Copy link
Copy Markdown
Contributor

Great perf improvement, lgtm

@chenweng-quic chenweng-quic changed the title Qualcomm AI Engine Direct - Change transpose output from NCH1 to NC1H before conv Qualcomm AI Engine Direct - Change transpose output from NCH1 to NC1W before conv Aug 12, 2026
@chenweng-quic
chenweng-quic force-pushed the dev1/chenweng/improve_prefill branch from fc5ddcd to cbeaa90 Compare August 12, 2026 02:06
@psiddh

psiddh commented Aug 12, 2026

Copy link
Copy Markdown
Contributor

psiddh added a commit that referenced this pull request Aug 12, 2026
…21765)

### Summary

The URL, xref, and file-size linters diffed `base..head`, where `base`
is the **tip** of the base branch rather than the point the branch left
it. A branch cut before recent commits still carries the lines those
commits replaced, so against the newer tip its old copies read as
additions, and the branch gets blamed for links someone else already
repaired.

#21729 failed exactly this way. It touches 18 files and adds no URLs at
all, but the two-dot diff scoped the lint to 143 files and flagged nine
links that #21694 had already fixed or `@lint-ignore`d. #21707 is
starker: one Python file with no URLs in it, 199 files linted, the same
nine failures.

```
base..head    143 files   <- what CI linted
base...head    18 files   <- what the PR actually changes
```

The stray `jq: parse error` lines in #21729's log are the same symptom
from the other direction: the job runs the branch's own pre-#21694 copy
of `lint_urls.sh`.

### Fix

The workflow resolves the merge base and passes it down. That is the
half that matters for branches already open: on a `pull_request` the
reusable workflow resolves from the merge commit, so it carries this fix
even though `scripts/` still comes from the branch itself. The scripts
also switch to three-dot ranges so `./scripts/lint_urls.sh main HEAD` by
hand behaves the same. `lint_xrefs.sh` and `lint_file_size.sh` had the
identical bug and get the identical change.

**Cost, stated plainly.** A merge base needs real history, so this
restores `fetch-depth: 0` on the `pull_request` path — line for line
what #17682 removed in February for speed. Measured on this branch, that
is ~25s of added wall clock and ~79s of runner time across the three
concurrent jobs. #17682's 6min → 10s was mostly the runner and Docker
change rather than the fetch depth, though the commit changes both at
once and I can't fully separate them. Pushes and the nightly whole-tree
scan cannot use a merge base and stay shallow:

```yaml
fetch-depth: ${{ github.event_name == 'pull_request' && '0' || '1' }}
```

The quotes matter — bare `0` is falsy in GitHub expressions, so `&& 0 ||
1` always yields `1` and would silently disable the fix.

**No silent fallback.** When there is no merge base there is no usable
range, so the range is left unset and the linters scan the whole tree.
Substituting the base tip instead produces a range the scripts fail on
quietly: verified against unrelated histories, `lint_urls.sh` and
`lint_xrefs.sh` both exit **0** having checked nothing, while
`lint_file_size.sh` exits 128 and the wrapper reports "some files exceed
the 1 MB limit", which is not what happened. Unset args are loud instead
— verified rc=1 on a tree containing a dead link.

**`--no-color`.** Both `git diff` calls now pass it, matching the `git
grep --no-color` two lines below. With `color.ui = always` in a
developer's config, added lines arrive wrapped in escape sequences,
`grep -E '^\+'` matches nothing, and the check passes having found
nothing (measured: 2 matches → 0).

### Test plan

`.ci/scripts/tests/test_link_check_diff_selection.py` builds a diverged
history where `main` repairs bad links and shrinks an oversized file
while the feature branch simply predates all of it. Four fixture files
each pin a different part, and `curl` is stubbed so there is no network:

| Fixture | Pins |
|---|---|
| `both_sides.md` — edited on both branches | the per-file diff range |
| `big.bin` — oversized at the branch point, shrunk on main |
`lint_file_size.sh`'s range |
| `colorful.gitconfig` — `color.ui = always` | `--no-color` |
| `base_only.md` / `feature_only.md` | that main-only changes stay
invisible and the branch's own additions are still checked |

```
pytest .ci/scripts/tests/test_link_check_diff_selection.py   # 4 passed
```

Mutating each changed line individually:

```
inner per-file range -> ..     3 failed   caught
lint_file_size range -> ..     1 failed   caught
drop --no-color                1 failed   caught
file-selection range -> ..     4 passed   equivalent mutant, see below
```

The file-selection range is not pinned because it cannot be: with the
per-file diff at three dots, the extra files it selects produce empty
diffs. Replayed against #21707's real 199-file range, both variants emit
byte-identical output. It is changed for consistency, not behaviour.

**Narrowing the scope must not blunt the check**, so each linter also
has a positive control where the branch itself adds the bad thing:

```
lint_urls      adds a dead URL         -> rc=1, reports example.invalid/dead
lint_xrefs     adds a broken reference -> rc=1, reports sub/missing.md
lint_file_size adds a 1MB+ file        -> rc=1, reports feature_big.bin
```

End to end on real content: a synthetic commit on top of `main` that
puts #21694's nine repaired links back, as if a PR had added them, gives
`rc=1` with 9 FAIL and 5 OK against the live network. Incidentally
`musl.cc` answered this time where CI saw `000`, and
`pybind/cmake_example` now 404s where CI saw `301` — which is the
retry-then-WARN path earning its keep.

Also replayed #21729's and #21707's exact CI refs through the fixed
scripts (exit 0 each), and ran all three linters plus `lintrunner`
against this PR's own diff.

### Known limitations

- The checkout is `head.sha`, so references still resolve against the
branch tree rather than the merge result — the mirror image of the bug
fixed here. Pre-existing and not worsened by this PR.
- The whole-tree branch swallows a `git grep` failure and exits 0, so on
a git built without PCRE the scan silently checks nothing. Reachable
locally, not on the runners, which is why the nightly scan works.
Pre-existing; worth a follow-up rather than widening this PR.

### Rollout

This does **not** repair a currently red run. A rerun keeps the original
`GITHUB_SHA`, and advancing the base alone does not fire `synchronize`,
so an already-open PR picks the fix up only on its next newly triggered
pull-request run — any push, or close/reopen. That is still cheaper than
a content rebase: no history rewrite, no conflicts, nothing to
re-review.

Supersedes #21762, which fixes the same bug but leaves the checkout
shallow and misses `lint_urls.sh`. Its reviewer's shallow-checkout point
is exactly right: a `--depth=1` fetch writes a shallow graft even into
an otherwise complete clone, after which `git merge-base` fails and
`A...B` is fatal, so the two halves of this change are a pair. The
regression test here is adapted from that PR.

Authored with Claude Code (Claude Opus 5).
@chenweng-quic
chenweng-quic force-pushed the dev1/chenweng/improve_prefill branch from cbeaa90 to 34b9aa6 Compare August 13, 2026 01:42
@chenweng-quic

Copy link
Copy Markdown
Collaborator Author

Can you check why test-qnn-testsuite-linux / test-backend-linux (qnn, models) / linux-job (pull_request) is failing ?

Hi @psiddh ,
it seems to a random failure.

@psiddh
psiddh merged commit df3d715 into pytorch:main Aug 13, 2026
184 checks passed
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 Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: qnn Issues related to Qualcomm's QNN delegate and code under backends/qualcomm/

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants