benchmarks/kernels/benchmark_reshape_and_cache.py · 16d2ad1d384e8b3d89434ae60508b26d0ce6ac99 · OpenDAS / vllm_cscc

[Hardware] Replace `torch.cuda.empty_cache` with `torch.accelerator.empty_cache` (#30681) · 16d2ad1d

Kunshang Ji authored Mar 04, 2026


Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

16d2ad1d

benchmark_reshape_and_cache.py 4.8 KB

Replace benchmark_reshape_and_cache.py