Add a fatal error message for unsupported GQA do_rotary attribute.

microsoft · Jan 8, 2025 · 73c563b · 73c563b
1 parent 4b0cee3
commit 73c563b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts b/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts
@@ -24,6 +24,9 @@ export const validateInputs = (
   inputs: readonly TensorView[],
   attributes: GroupQueryAttentionAttributes,
 ): AttentionParameters => {
+  if (attributes.doRotary) {
+    throw new Error('GroupQuerryAttention do_rotary attribute is not supported');
+  }
   if (attributes.doRotary && inputs.length <= 7) {
     throw new Error('cos_cache and sin_cache inputs are required if do_rotary is specified');
   }