put_record_event_in_python_on_timeline_python (#42555) (#42790)

* put_record_event_in_python_on_timeline_python * fix
PaddlePaddle · May 17, 2022 · a40e60f · a40e60f
1 parent 07029e0
commit a40e60f
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/python/paddle/profiler/utils.py b/python/paddle/profiler/utils.py
@@ -25,9 +25,9 @@
 
 _AllowedEventTypeList = [
     TracerEventType.Dataloader, TracerEventType.ProfileStep,
-    TracerEventType.UserDefined, TracerEventType.Forward,
-    TracerEventType.Backward, TracerEventType.Optimization,
-    TracerEventType.PythonOp, TracerEventType.PythonUserDefined
+    TracerEventType.Forward, TracerEventType.Backward,
+    TracerEventType.Optimization, TracerEventType.PythonOp,
+    TracerEventType.PythonUserDefined
 ]
 
 
@@ -37,7 +37,7 @@ class RecordEvent(ContextDecorator):
 
     Args:
         name(str): Name of the record event
-        event_type(TracerEventType, optional): Optional, default value is TracerEventType.UserDefined. It is reserved for internal purpose, and it is better not to specify this parameter. 
+        event_type(TracerEventType, optional): Optional, default value is TracerEventType.PythonUserDefined. It is reserved for internal purpose, and it is better not to specify this parameter. 
 
     Examples:
         .. code-block:: python
@@ -64,7 +64,7 @@ class RecordEvent(ContextDecorator):
 
     def __init__(self,
                  name: str,
-                 event_type: TracerEventType=TracerEventType.UserDefined):
+                 event_type: TracerEventType=TracerEventType.PythonUserDefined):
         self.name = name
         self.event_type = event_type
         self.event = None
@@ -101,8 +101,6 @@ def begin(self):
                   can be recorded.".format(*_AllowedEventTypeList))
             self.event = None
         else:
-            if self.event_type == TracerEventType.UserDefined:
-                self.event_type == TracerEventType.PythonUserDefined
             self.event = _RecordEvent(self.name, self.event_type)
 
     def end(self):