[XPU] Alias is_current_stream_capturing to XPU in cuda wrapper (#50526)
Signed-off-by: Sundaresan-G <sundaresan.g@intel.com> Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
Sundaresan G committed
Aug 5, 2026 at 12:40 UTC
999dd8b49031a7bc0f98a0fe88a1ddc2ead6a754
1 file changed
+1
vllm/v1/worker/xpu_model_runner.py
+1
@@ -60,4 +60,5 @@ def _torch_cuda_wrapper():
60
torch.cuda.graph = partial(torch.xpu.graph)
61
torch.cuda.CUDAGraph = torch.xpu.XPUGraph
62
torch.cuda.graph_pool_handle = partial(torch.xpu.graph_pool_handle)
63
+ torch.cuda.is_current_stream_capturing = torch.xpu.is_current_stream_capturing
64
yield