[CI/Build] Share nightly CUDA YAML between H100/L4 and B200 (#7028)

Signed-off-by: wangyu <410167048@qq.com>
Signed-off-by: [Your Name] [Your Email]
Signed-off-by: [Your Name] <your.email@example.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: amy-why-3459 <wuhaiyan17@huawei.com>

tests/examples/offline_inference/test_image_to_video.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: Apache-2.0
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
+
 """
 Offline inference tests: image-to-video.
 See examples/offline_inference/image_to_video/README.md
@@ -17,7 +20,7 @@ pytestmark = [
     pytest.mark.usefixtures("clean_gpu_memory_between_tests"),
     pytest.mark.full_model,
     pytest.mark.example,
-    *hardware_marks(res={"cuda": "H100"}),
+    *hardware_marks(res={"cuda": ["H100", "B200"]}),
 ]
 
 I2V_SCRIPT = EXAMPLES / "offline_inference" / "image_to_video" / "image_to_video.py"

tests/examples/offline_inference/test_internvla_a1.py

@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
 
 """Offline E2E test for InternVLA-A1 open-loop inference."""
 
@@ -26,6 +26,7 @@ def _required_env(name: str) -> str:
     value = os.getenv(name)
     if not value:
         pytest.skip(f"{name} is not set")
+    assert value
     return value
 
 

tests/examples/offline_inference/test_text_to_image.py

@@ -18,7 +18,7 @@ pytestmark = [
     pytest.mark.usefixtures("clean_gpu_memory_between_tests"),
     pytest.mark.full_model,
     pytest.mark.example,
-    *hardware_marks(res={"cuda": "H100"}),
+    *hardware_marks(res={"cuda": ["H100", "B200"]}),
 ]
 
 T2I_SCRIPT = EXAMPLES / "offline_inference" / "text_to_image" / "text_to_image.py"

tests/examples/offline_inference/test_vace_video_generation.py

@@ -1,3 +1,6 @@
+# SPDX-License-Identifier: Apache-2.0
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
+
 """VACE coverage through the shared text-to-video and image-to-video examples."""
 
 from pathlib import Path
@@ -14,7 +17,7 @@ pytestmark = [
     pytest.mark.full_model,
     pytest.mark.diffusion,
     pytest.mark.example,
-    *hardware_marks(res={"cuda": "H100"}),
+    *hardware_marks(res={"cuda": ["H100", "B200"]}),
 ]
 
 TEXT_TO_VIDEO_README = EXAMPLES / "offline_inference" / "text_to_video" / "text_to_video.md"