[None][feat] add waive by sm version (#8928)

Signed-off-by: Xin He (SW-GPU) <[email protected]>

Author: xinhe-nv

tests/integration/README.md

@@ -17,6 +17,18 @@ The following subfolder contains test definitions for Tensorrt LLM.
         └── waives.txt  #         Test waive list
 ~~~
 
+## Test Waives
+
+The `waives.txt` file supports skipping tests based on:
+- **Platform name**: e.g., `full:RTX/`, `full:DGX-A100-40GB/`, `full:GH200/`
+- **SM version**: e.g., `full:sm90/`, `full:sm89/`, `full:sm100/`
+
+SM version mapping:
+- `sm89` = Ada (e.g., RTX 4090, L40S)
+- `sm90` = Hopper (e.g., H100, H200)
+- `sm100` = Blackwell (e.g., B100, B200)
+- `sm103` = Blackwell-Ultra (e.g., B300, GB300)
+
 - To run perf tests, you also need to first build the cpp benchmark by calling `build_wheel.py` with `--benchmarks` flag.
 
 ## Run perf tests

tests/integration/defs/test_list_parser.py

@@ -102,7 +102,22 @@ def parse_test_name(s):
         if s.startswith("full:"):
             s = s.lstrip("full:")
             if test_prefix:
-                if test_prefix.split('-')[0] in s:
+                # Check for SM version pattern (e.g., sm90, sm89, sm100)
+                sm_match = re.match(r'sm(\d+)/', s)
+                if sm_match:
+                    sm_version = int(sm_match.group(1))
+                    # Get current SM version
+                    try:
+                        from .conftest import get_sm_version
+                        current_sm = get_sm_version()
+                        # If SM versions match, replace with test_prefix
+                        if sm_version == current_sm:
+                            s = s.replace(f'sm{sm_version}',
+                                          test_prefix.split('-')[0])
+                    except Exception:
+                        # If can't get SM version, skip SM-based filtering
+                        pass
+                elif test_prefix.split('-')[0] in s:
                     s = s.replace(test_prefix.split('-')[0], test_prefix)
             return s
         elif test_prefix:

tests/integration/defs/triton_server/test_list_parser.py

@@ -103,7 +103,22 @@ def parse_test_name(s):
         if s.startswith("full:"):
             s = s.lstrip("full:")
             if test_prefix:
-                if test_prefix.split('-')[0] in s:
+                # Check for SM version pattern (e.g., sm90, sm89, sm100)
+                sm_match = re.match(r'sm(\d+)/', s)
+                if sm_match:
+                    sm_version = int(sm_match.group(1))
+                    # Get current SM version
+                    try:
+                        from ..conftest import get_sm_version
+                        current_sm = get_sm_version()
+                        # If SM versions match, replace with test_prefix
+                        if sm_version == current_sm:
+                            s = s.replace(f'sm{sm_version}',
+                                          test_prefix.split('-')[0])
+                    except Exception:
+                        # If can't get SM version, skip SM-based filtering
+                        pass
+                elif test_prefix.split('-')[0] in s:
                     s = s.replace(test_prefix.split('-')[0], test_prefix)
             return s
         elif test_prefix: