diff --git a/test/profiler/test_profiler.py b/test/profiler/test_profiler.py
index b204fbfcd227f0..777f3e988149b2 100644
--- a/test/profiler/test_profiler.py
+++ b/test/profiler/test_profiler.py
@@ -2012,7 +2012,7 @@ def test_dynamic_toggle(self):
 
         self.assertTrue(any("aten" in e.name for e in p.events()))
 
-        self.assertTrue(any("cuda" in e.name for e in p.events()))
+        self.assertTrue(any("cuda" in e.name for e in p.events()) or any("hip" in e.name for e in p.events()))
 
         self.assertTrue(any("kernel" in e.name for e in p.events()))