diff --git a/python/paddle/profiler/profiler.py b/python/paddle/profiler/profiler.py
index eef6714f2e20c94421587382607260ea678e3e6d..e274f007b83b0ec78134f645b316f2080e775e69 100644
--- a/python/paddle/profiler/profiler.py
+++ b/python/paddle/profiler/profiler.py
@@ -31,17 +31,14 @@ from paddle.fluid.core import (
     enable_memory_recorder,
     enable_op_info_recorder,
 )
+from paddle.profiler import utils
 
-from .utils import RecordEvent, wrap_optimizers
 from .profiler_statistic import (
+    SortedKeys,
     StatisticData,
     _build_table,
-    SortedKeys,
     gen_layer_flops,
 )
-from paddle.profiler import utils
-
-from .profiler_statistic import SortedKeys, StatisticData, _build_table
 from .timer import benchmark
 from .utils import RecordEvent, wrap_optimizers
 
@@ -359,6 +356,7 @@ class Profiler:
             be timed and profiled. Default: False.
         record_shapes (bool, optional): If it is True, collect op's input shape information. Default: False.
         profile_memory (bool, optional): If it is True, collect tensor memory allocation and release information. Default: False.
+        custom_device_types (list, optional): If targets contain profiler.ProfilerTarget.CUSTOM_DEVICE, custom_device_types select the custom device type for profiling. The default value represents all custom devices will be selected.
         with_flops (bool, optional): If it is True, the flops of the op will be calculated. Default: False.
 
     Examples: