apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 08:21:53 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[XPU] Fix wrapper files (#6830)

Browse Source 
- Add WRAPPER_CHECK_PTR for pointer validity checks
- Add WRAPPER_ASSERT_GT/GE/LE for parameter range validation
- Simplify wrapper function calls to direct return pattern
This commit is contained in:
mayang002
2026-03-16 14:39:40 +08:00
committed by GitHub
parent 3fabba0dc7
commit 72ff7bf4cd
15 changed files with 165 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/xpu_ops/src/plugin/include/xpu/plugin.h
+2 -1
View File
@@ -77,7 +77,8 @@ DLL_EXPORT int get_padding_offset(api::Context* ctx,
const int* cum_offsets,
const int* seq_lens,
const int max_seq_len,
const int bs);
const int bs,
const int64_t token_num);
DLL_EXPORT int speculate_get_padding_offset(api::Context* ctx,
int* batch_id_per_token,