From fba04e857b57abc54ba4864cbfb3af0461e2c5e7 Mon Sep 17 00:00:00 2001
From: genghaozhe <939857490@qq.com>
Date: Sat, 25 May 2024 14:55:09 +0000
Subject: [PATCH] [bugs] fix args.profile=False DummyProfiler errro

---
 examples/language/performance_evaluator.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/examples/language/performance_evaluator.py b/examples/language/performance_evaluator.py
index 0b147b7ea..99df8f1da 100644
--- a/examples/language/performance_evaluator.py
+++ b/examples/language/performance_evaluator.py
@@ -36,6 +36,12 @@ def get_profile_context(enable_flag, warmup_steps, active_steps, save_dir):
         def step(self):
             self.step_number += 1
 
+        def __enter__(self):
+            return self
+
+        def __exit__(self, exc_type, exc_value, traceback):
+            pass
+
     if enable_flag:
         return profile(
             activities=[ProfilerActivity.CPU, ProfilerActivity.CUDA],