Skip to content

Commit

Permalink
[profiler ut] Update event name in test_profiler.py (pytorch#131757)
Browse files Browse the repository at this point in the history
Fixes #ISSUE_NUMBER
To support kernel name with some uppercase letters.
Pull Request resolved: pytorch#131757
Approved by: https://github.com/aaronenyeshi
  • Loading branch information
fwenguang authored and pytorchmergebot committed Jul 30, 2024
1 parent 32c57e7 commit 3864a2d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/profiler/test_profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -519,9 +519,9 @@ def test_kineto(self):
for e in p.function_events:
if "aten::mm" in e.name:
found_mm = True
if "gemm" in e.name or "Cijk" in e.name:
if "gemm" in e.name.lower() or "Cijk" in e.name:
found_gemm = True
if "Memcpy" in e.name or "memcpy" in e.name:
if "memcpy" in e.name.lower():
found_memcpy = True
if use_cuda:
self.assertTrue(found_gemm)
Expand Down

0 comments on commit 3864a2d

Please sign in to comment.