We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 72b02ec commit 1f38016Copy full SHA for 1f38016
build/packaging/smoke_test.py
@@ -85,8 +85,8 @@ def main():
85
86
# Make sure custom ops are registered.
87
assert (
88
- "llama::sdpa_with_kv_cache" in ops
89
- ), f"sdpa_with_kv_cache not registered, Got ops: {ops}"
+ "llama::sdpa_with_kv_cache.out" in ops
+ ), f"llama::sdpa_with_kv_cache.out not registered, Got ops: {ops}"
90
91
# Make sure quantized ops are registered.
92
0 commit comments