Merge pull request #183 from arghdos/fix_for_dispatch_selection_of_st…

…ochastic fix for stoichastic kernel dispatch selection
ROCm · Sep 11, 2023 · 6cc220d · 6cc220d
2 parents 1289682 + a0955c2
commit 6cc220d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/omniperf_analyze/utils/parser.py b/src/omniperf_analyze/utils/parser.py
@@ -763,11 +763,8 @@ def apply_filters(workload, is_gui, debug):
                 ret_df[schema.pmc_perf_file_prefix]["Index"] > int(m.group(1))
             ]
         else:
-            ret_df = ret_df.loc[
-                ret_df[schema.pmc_perf_file_prefix]["Index"]
-                .astype(str)
-                .isin(workload.filter_dispatch_ids)
-            ]
+            dispatches = [int(x) for x in workload.filter_dispatch_ids]
+            ret_df = ret_df.loc[dispatches]
     if debug:
         print("~" * 40, "\nraw pmc df info:\n")
         print(workload.raw_pmc.info())