Skip to content

[OOB] use sdpa fusion in eager models #3778

Open

Description

@jianyizh

change and upstream model script to enable sdpa in the following models in inference. Torch compile has fused them by pattern matcher.

# Model
1 volo_d1_224_bs2048
2 tnt_s_patch16_224_bs2048
3 LayoutLMForSequenceClassification_bs256
4 hf_T5_large_bs32
5 hf_T5_bs16
6 T5ForConditionalGeneration_bs32
7 T5Small_bs32
8 hf_T5_generate_bs16
9 visformer_small_bs2048
10 hf_T5_base_bs1
11 MT5ForConditionalGeneration_bs64

Summary: 11 model(s) could use F.scaled_dot_product_attention in eager code.

Activity

  1. added this to the PT2.15 milestone on Jun 3, 2026
  2. modified the milestones: PT2.15, PT2.16 on Sep 23, 2026
  3. added
    module: sdpaSDPA / flash / efficient attention kernels
    inferenceInference workload phase
    on Nov 11, 2026
  4. added theissue type on Nov 11, 2026
  5. daisyden commented on Nov 11, 2026

    @daisyden
    Contributor
    label-issue: #3778

    Root cause: enablement task, not a failure — 11 eager models could use F.scaled_dot_product_attention (already fused by torch.compile pattern matcher); requires upstreaming model-script changes. Owner: model scripts / torch-xpu-ops OOB.

    axis value reason
    type Feature No native Type; body asks to "enable sdpa" in eager models, a new capability.
    test test: oob Label test: oob; [OOB] out-of-box eager model suite.
    module module: sdpa Task is to adopt scaled_dot_product_attention (SDPA) fusion.
    priority Low Enhancement/enablement request, no failure.
    symptom inference Models targeted in the inference phase per body.

    Applied to this issue. Generated by a local label-issue run.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

inferenceInference workload phasemodule: sdpaSDPA / flash / efficient attention kernelstest: oobOut-of-box model test failure surface

Projects

No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions