apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[XPU] fix text_image_gather_scatter when image_token_num == token_num && text_token_num == 1 (#4881)

Browse Source
This commit is contained in:
Lucas
2025-11-07 16:35:49 +08:00
committed by GitHub
parent 71bbedaf50
commit 3b0bdbae65
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
custom_ops/xpu_ops/src/plugin/src/wrapper/text_image_gather_scatter.cpp
View File

@@ -169,7 +169,11 @@ int text_image_gather_scatter(
WRAPPER_CHECK_PTR(ctx, int, token_num, token_type_ids);
WRAPPER_CHECK_PTR(ctx, int, token_num, text_index);
WRAPPER_CHECK_PTR(ctx, int, token_num, image_index);
WRAPPER_ASSERT_EQ(ctx, token_num, text_token_num + image_token_num);
// When all tokens are image type, text_token_num will be set to 1 in model.py
WRAPPER_ASSERT_EQ(ctx,
true,
(text_token_num + image_token_num == token_num) ||
(image_token_num == token_num && text_token_num == 1));
if (ctx->dev().type() == api::kCPU) {
return cpu_wrapper<T>(ctx,