iHore/vLLM
8
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[TEST][XPU][CI] Fix ci uva kernel test instability 2 (#60404)

Browse Source 
Signed-off-by: microslaw <milosz.grunwald@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
This commit is contained in:
Miłosz Grunwald
2026-10-10 12:09:13 +08:00
committed by GitHub
co-authored by Kunshang Ji
parent f1a3c60558
commit be0823357f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/kernels/core/test_uva.py
+6
View File
@@ -21,6 +21,9 @@ DEVICES = [
@pytest.mark.parametrize("device", DEVICES)
def test_cpu_write(device):
torch.set_default_device(device)
# Initializes the device; otherwise torch.accelerator.synchronize() is a no-op,
# as the UVA view's ops never trigger lazy init.
torch.accelerator.set_device_index(device)
cpu_tensor = torch.zeros(10, 10, device="cpu", pin_memory=True, dtype=torch.int32)
gpu_view = get_accelerator_view_from_cpu_tensor(cpu_tensor)
assert gpu_view.device.type == DEVICE_TYPE
@@ -44,6 +47,9 @@ def test_cpu_write(device):
@pytest.mark.parametrize("device", DEVICES)
def test_gpu_write(device):
torch.set_default_device(device)
# Initializes the device; otherwise torch.accelerator.synchronize() is a no-op,
# as the UVA view's ops never trigger lazy init.
torch.accelerator.set_device_index(device)
cpu_tensor = torch.zeros(10, 10, device="cpu", pin_memory=True, dtype=torch.int32)
gpu_view = get_accelerator_view_from_cpu_tensor(cpu_tensor)
assert gpu_view.device.type == DEVICE_TYPE