enable_operator_stats_collection¶
启用以收集不同数据类型的算子调用次数。按照 float32、float16、bfloat16 四种数据类型统计算子调用次数, 此函数与相应的禁用函数配对使用。
返回¶
无返回值
代码示例¶
>>> import paddle
>>> paddle.device.set_device('gpu')
>>> 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 | 0 | 0 | 1 | 0
>>> # reshape2 | 0 | 0 | 1 | 0
>>> # transfer_dtype | 1 | 0 | 2 | 0
>>> # <----------------------------------------------------- op count: 4 ------------------------------------------------------>