pnnx handles sdpa batch index (#5617)

Tencent · Aug 8, 2024 · 60823a8 · 60823a8
1 parent 03cf161
commit 60823a8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/pnnx/src/pass_ncnn/solve_batch_index.cpp b/tools/pnnx/src/pass_ncnn/solve_batch_index.cpp
@@ -56,6 +56,7 @@ static bool is_known_operator_with_batch_index_0(const Operator* op)
         "F.pixel_shuffle",
         "F.pixel_unshuffle",
         "F.prelu",
+        "F.scaled_dot_product_attention",
         "F.unfold",
         "F.upsample_bilinear",
         "F.upsample_nearest",