[rocprofiler-compute] Skip PC sampling test if not supported on machine post profiling (#4387)

vedithal-amd · web-flow · commit 4dfb4797f07c · 2026-03-25T01:28:39.000-04:00
## Motivation CI machine can sometimes run on machine which does not support stochastic pc sampling driver/firmware Fixes ROCm/TheRock#4034  ## Technical Details In test_pc_sampling.py skip stochastic pc sampling test if profiling output complains about PC sampling configuration not supported  ## JIRA ID   ## Test Plan Run test_pc_sampling.py locally  ## Test Result Tests pass  ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
diff --git a/projects/rocprofiler-compute/tests/test_pc_sampling.py b/projects/rocprofiler-compute/tests/test_pc_sampling.py
@@ -61,6 +61,14 @@
 ])
 
 
+def is_pc_sampling_not_supported(output):
+    """
+    To be called with the stdout + stderr after profiling.
+    Check whether profiling output said PC sampling is not supported on the machine
+    """
+    return "Given PC sampling configuration is not supported" in output
+
+
 def test_pc_sampling_host_trap(binary_handler_profile_rocprof_compute):
     """
     Test that PC sampling works with --block 21 and --pc-sampling-method host_trap.
@@ -114,15 +122,22 @@ def test_pc_sampling_stochastic(binary_handler_profile_rocprof_compute):
 
     workload_dir = test_utils.get_output_dir()
 
-    _ = binary_handler_profile_rocprof_compute(
+    code, stdout, stderr = binary_handler_profile_rocprof_compute(
         config,
         workload_dir,
         options,
-        check_success=True,
+        check_success=False,
+        capture_output=True,
         roof=False,
         app_name="app_mat_mul_max",
     )
 
+    output = f"{stdout}\n{stderr}"
+    if is_pc_sampling_not_supported(output):
+        test_utils.clean_output_dir(config["cleanup"], workload_dir)
+        pytest.skip("PC sampling is not supported")
+
+    assert code == 0
     file_dict = test_utils.check_non_pmc_files(workload_dir, num_devices, 1)
     assert sorted(list(file_dict.keys())) == sorted(PC_SAMPLING_STOCHASTIC_FILES)
 
