8000 scaled_dot_product_attention(): argument 'is_causal' must be bool, not SymBool · Issue #154038 · pytorch/pytorch · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
scaled_dot_product_attention(): argument 'is_causal' must be bool, not SymBool #154038
New issue
New issue
Closed as not planned
Closed as not planned
scaled_dot_product_attention(): argument 'is_causal' must be bool, not SymBool#154038
Labels
oncall: exportoncall: pt2
@titaiwangms

Description

@titaiwangms
titaiwangms
opened on May 21, 2025

pytorch/aten/src/ATen/native/native_functions.yaml

Line 14877 in 72a3c8d

- func: scaled_dot_product_attention(Tensor query, Tensor key, Tensor value, Tensor? attn_mask=None, float dropout_p=0.0, bool is_causal=False, *, float? scale=None, bool enable_gqa=False) -> Tensor 

# Next ones fails
#   File "python3.12/site-packages/whisper/model.py", line 124, in qkv_attention
#   a = scaled_dot_product_attention(
#       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# TypeError: scaled_dot_product_attention(): argument 'is_causal' must be bool, not SymBool

cc @chauhang @penguinwu @avikchaudhuri @gmagogsfm @zhxchen17 @tugsbayasgalan @angelayi @suo @ydwu4

Metadata

Metadata

Assignees

No one assigned

    Labels

    oncall: exportoncall: pt2

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions
      0