From 5d19da5db4e0ecb95107c9629d3e77600653a9bc Mon Sep 17 00:00:00 2001 From: Giovanni LB Date: Tue, 20 Feb 2024 16:52:12 -0300 Subject: [PATCH] SWDEV-446854: Fixing ATT crash on rocblas_sgemm Change-Id: I932b92c4e188655500b1b150d9594ec663dd25c7 [ROCm/rocprofiler commit: a1999fa366390e96439ab5032736904034d49229] --- projects/rocprofiler/plugin/att/stitch.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/projects/rocprofiler/plugin/att/stitch.py b/projects/rocprofiler/plugin/att/stitch.py index 4fd2128d78..bbc2c6f555 100644 --- a/projects/rocprofiler/plugin/att/stitch.py +++ b/projects/rocprofiler/plugin/att/stitch.py @@ -455,7 +455,10 @@ def stitch(insts, raw_code, jumps, gfxv, bIsAuto, codeservice): NUM_FLAT += 1 elif inst.type == IMMED and "s_waitcnt" in as_line[0]: if "lgkmcnt" in as_line[0]: - wait_N = int(as_line[0].split("lgkmcnt(")[1].split(")")[0]) + try: + wait_N = int(as_line[0].split("lgkmcnt(")[1].split(")")[0]) + except: + wait_N = 0 flight_count.append([as_line[5], num_inflight, wait_N]) if wait_N == 0: smem_ordering = 0 @@ -474,7 +477,10 @@ def stitch(insts, raw_code, jumps, gfxv, bIsAuto, codeservice): num_inflight = NUM_FLAT + NUM_SMEM + NUM_VLMEM + NUM_VSMEM if "vmcnt" in as_line[0]: - wait_N = int(as_line[0].split("vmcnt(")[1].split(")")[0]) + try: + wait_N = int(as_line[0].split("vmcnt(")[1].split(")")[0]) + except: + wait_N = 0 flight_count.append([as_line[5], num_inflight, wait_N]) if wait_N == 0: vlmem_ordering = 0