o ۷i@s$ddlmZddlZeddZdS))contextmanagerNccsBztjjtj_tjjtj_tjjtj_tjjtj_dVWdSw)N)torchxpuStreamcudacurrent_streamdefault_streamstreamr r S/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/platforms/xpu/utils.pytorch_cuda_wrappers    r ) contextlibrrr r r r r s