Skip to content

Commit d969c30

Browse files
ronny1996Ligoml
andauthored
add custom profier doc (#48629)
* add custom profier doc * update * format; test=document_fix Co-authored-by: Ligoml <[email protected]>
1 parent 493825a commit d969c30

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

python/paddle/profiler/profiler.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,17 +31,14 @@
3131
enable_memory_recorder,
3232
enable_op_info_recorder,
3333
)
34+
from paddle.profiler import utils
3435

35-
from .utils import RecordEvent, wrap_optimizers
3636
from .profiler_statistic import (
37+
SortedKeys,
3738
StatisticData,
3839
_build_table,
39-
SortedKeys,
4040
gen_layer_flops,
4141
)
42-
from paddle.profiler import utils
43-
44-
from .profiler_statistic import SortedKeys, StatisticData, _build_table
4542
from .timer import benchmark
4643
from .utils import RecordEvent, wrap_optimizers
4744

@@ -359,6 +356,7 @@ class Profiler:
359356
be timed and profiled. Default: False.
360357
record_shapes (bool, optional): If it is True, collect op's input shape information. Default: False.
361358
profile_memory (bool, optional): If it is True, collect tensor memory allocation and release information. Default: False.
359+
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.
362360
with_flops (bool, optional): If it is True, the flops of the op will be calculated. Default: False.
363361
364362
Examples:

0 commit comments

Comments
 (0)