test(performance): align flat recipe unit coverage

Signed-off-by: yaoyu-33 <yaoyu.094@gmail.com>

Author: yaoyu-33

tests/unit_tests/recipes/test_override_precedence.py

@@ -23,12 +23,23 @@
 
 import sys
 from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+import pytest
 
 
 SCRIPTS_PERF_PATH = Path(__file__).parents[3] / "scripts" / "performance"
 sys.path.insert(0, str(SCRIPTS_PERF_PATH))
 
 
+@pytest.fixture(autouse=True)
+def _mock_cuda_device_properties():
+    """Keep recipe construction independent of the test host's GPUs."""
+    properties = MagicMock(major=9, name="NVIDIA H100")
+    with patch("torch.cuda.get_device_properties", return_value=properties):
+        yield
+
+
 def _build_base_args(**overrides):
     """Return a fully-populated argparse.Namespace using the real parser defaults.
 

tests/unit_tests/scripts/performance/test_perf_plugins.py

@@ -46,7 +46,6 @@ def test_set_determinism_env_vars_writes_three_keys():
         gpu="h100",
         compute_dtype="bf16",
         train_task="pretrain",
-        num_gpus=64,
     )
     executor = MagicMock()
     executor.env_vars = {}
@@ -56,15 +55,3 @@ def test_set_determinism_env_vars_writes_three_keys():
     assert executor.env_vars["NCCL_ALGO"] == "Ring"
     assert executor.env_vars["NVTE_ALLOW_NONDETERMINISTIC_ALGO"] == "0"
     assert executor.env_vars["CUBLAS_WORKSPACE_CONFIG"] == ":4096:8"
-
-
-def test_perf_env_plugin_num_gpus_is_optional():
-    plugin = PerfEnvPlugin(
-        model_family_name="llama",
-        model_recipe_name="llama3_70b",
-        gpu="h100",
-        compute_dtype="bf16",
-        train_task="pretrain",
-    )
-
-    assert plugin.num_gpus is None