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

[CI] Add unit test for swap_layout && remove unit test of splitwise_scheduler (#6250)

Browse Source 
* Add unit test for swap_layout

* remove splitwise_scheduler test
This commit is contained in:
jc
2026-01-28 19:20:20 +08:00
committed by GitHub
parent 6d33d5e370
commit 7da5f54fb3
3 changed files with 97 additions and 428 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/gpu_ops/swap_cache_layout.cu
-1
View File
@@ -53,7 +53,6 @@ void SwapCacheImpLayout(
const paddle::Tensor& cache_gpu = cache_gpu_tensors[layer_idx];
data_t* cache_gpu_ptr = const_cast<data_t*>(cache_gpu.data<data_t>());
auto* cache_cpu_ptr = reinterpret_cast<data_t*>(cache_cpu_pointer);
// auto stream = cache_gpu.stream();
for (int block_idx = 0; block_idx < gpu_block_ids.size(); block_idx++) {
auto cur_gpu_block_id = gpu_block_ids[block_idx];