fix(runner): fix timer args dict

jinmingyi1998 · Aug 9, 2023 · 9d8adbe · 9d8adbe
1 parent d2e89ae
commit 9d8adbe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/oclk/oclk_runner.py b/oclk/oclk_runner.py
@@ -197,12 +197,17 @@ def run(
         :rtype: TimerResult
         """
         assert kernel_name in self.kernel_list
+        timer_dict:Dict = {}
+        if isinstance(timer,TimerArgs):
+            timer_dict = timer.__dict__()
+        else:
+            timer_dict = timer
         return F.run(
             kernel_name=kernel_name,
             input=input,
             output=output,
             local_work_size=local_work_size,
             global_work_size=global_work_size,
             wait=wait,
-            timer=timer,
+            timer=timer_dict,
         )