Skip to content

Commit

Permalink
Remove test that only works with amdgpu driver
Browse files Browse the repository at this point in the history
  • Loading branch information
jordap committed Jun 27, 2024
1 parent a3cb124 commit cb6babb
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions test/test_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,17 +46,6 @@ def test_query_rocm(self):
_, value = results[0]["value"]
assert int(value) >= 0, "Reported power is too low"

@pytest.mark.skipif(rocm_host, reason="requires no GPUs in the host")
def test_query_gpus(self):
prometheus = PrometheusConnect(url=self.url)
query = f"rocm_num_gpus{{instance='{self.node}'}}"
results = prometheus.custom_query(query)
assert len(results) >= 1, "Metric rocm_num_gpus not available"

results = prometheus.custom_query(query)
_, value = results[0]["value"]
assert int(value) == 0, "There should be no GPUs"

def test_job(self):
prometheus = PrometheusConnect(url=self.url)
query = f"slurmjob_info{{jobid=~'.+'}}[{self.time_range}]"
Expand Down

0 comments on commit cb6babb

Please sign in to comment.