(tvm) branch main updated: [Tests][Relax] Gate multi-GPU VM test on three devices (#19725)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new a1d46dc5ab [Tests][Relax] Gate multi-GPU VM test on three devices 
(#19725)
a1d46dc5ab is described below

commit a1d46dc5ab77d33991d20cc3f6ef1486a8377a46
Author: Shushi Hong <820958...@qq.com>
AuthorDate: Wed Jun 10 18:26:50 2026 -0400

    [Tests][Relax] Gate multi-GPU VM test on three devices (#19725)
    
    This pr gates the Relax multi-GPU VM test on the actual number of
    visible CUDA devices.
---
 tests/python/relax/test_vm_multi_device.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/python/relax/test_vm_multi_device.py 
b/tests/python/relax/test_vm_multi_device.py
index d065d0078e..dbbf8a7014 100644
--- a/tests/python/relax/test_vm_multi_device.py
+++ b/tests/python/relax/test_vm_multi_device.py
@@ -17,6 +17,7 @@
 """Test eliminate common subexpr pass"""
 
 import numpy as np
+import pytest
 
 import tvm
 import tvm.testing
@@ -87,6 +88,9 @@ def test_multi_cpu():
 
 @tvm.testing.requires_multi_gpu
 def test_multi_gpu():
+    if not tvm.cuda(2).exist:
+        pytest.skip("requires at least 3 visible CUDA devices")
+
     @I.ir_module
     class Example:
         I.module_attrs({"attr": 10})