enable_operator_stats_collection¶

paddle.amp.debugging. enable_operator_stats_collection ( ) [源代码] ¶

启用以收集不同数据类型的算子调用次数。按照 float32、float16、bfloat16 四种数据类型统计算子调用次数, 此函数与相应的禁用函数配对使用。

返回¶

无返回值

代码示例¶

          import paddle

conv = paddle.nn.Conv2D(3, 2, 3)
x = paddle.rand([10, 3, 32, 32])

paddle.amp.debugging.enable_operator_stats_collection()
# AMP list including conv2d, elementwise_add, reshape2, cast (transfer_dtype)
with paddle.amp.auto_cast(enable=True, level='O2'):
    out = conv(x)
# Print to the standard output.
paddle.amp.debugging.disable_operator_stats_collection()
# <------------------------------------------------------- op list -------------------------------------------------------->
# <--------------- Op Name ---------------- | -- FP16 Calls --- | -- BF16 Calls --- | --- FP32 Calls--- | -- Other Calls -->
#   conv2d                                  |  1                |  0                |  0                |  0
#   elementwise_add                         |  1                |  0                |  0                |  0
#   reshape2                                |  1                |  0                |  0                |  0
#   transfer_dtype                          |  0                |  0                |  3                |  0
# <----------------------------------------------------- op count: 4 ------------------------------------------------------>

         