From ecce54e5b84135a2c63e901402885ed59d3a9005 Mon Sep 17 00:00:00 2001 From: sriram kuamr k Date: Wed, 18 Jun 2025 00:59:08 -0700 Subject: [PATCH] =?UTF-8?q?For=20profiling=20cuda=20code=20on=20rocm=20dev?= =?UTF-8?q?ice,=20we=20need=20to=20check=20for=20presence=20of=20=E2=80=9C?= =?UTF-8?q?hip=E2=80=9D=20(hipified=20code)=20instead=20of=20=E2=80=9Ccuda?= =?UTF-8?q?=E2=80=9D=20in=20the=20profiler=20event=20names.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/profiler/test_profiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/profiler/test_profiler.py b/test/profiler/test_profiler.py index b204fbfcd227f..777f3e988149b 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()))