Skip to content

Commit 1f38016

Browse files
authored
Fix op not found issue in smoke test for pip wheel (#6549)
As titled. Fixing https://github.com/pytorch/executorch/actions/runs/11573027936/job/32214255958
1 parent 72b02ec commit 1f38016

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

build/packaging/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,8 @@ def main():
8585

8686
# Make sure custom ops are registered.
8787
assert (
88-
"llama::sdpa_with_kv_cache" in ops
89-
), f"sdpa_with_kv_cache not registered, Got ops: {ops}"
88+
"llama::sdpa_with_kv_cache.out" in ops
89+
), f"llama::sdpa_with_kv_cache.out not registered, Got ops: {ops}"
9090

9191
# Make sure quantized ops are registered.
9292
assert (

0 commit comments

Comments
 (0)