diff --git a/tb_plugin/torch_tb_profiler/profiler/tensor_core.py b/tb_plugin/torch_tb_profiler/profiler/tensor_core.py index 9b5df3837..3a69cf70b 100644 --- a/tb_plugin/torch_tb_profiler/profiler/tensor_core.py +++ b/tb_plugin/torch_tb_profiler/profiler/tensor_core.py @@ -26,8 +26,7 @@ def __contains__(cls, item): class TC_OP_Allowlist(metaclass=TC_Allowlist_Meta): # Refer to https://github.com/pytorch/pytorch/blob/69b2bf70f9c0e591ce5e566afa59e19618031ead/aten/src/ATen/autocast_mode.cpp#L290-L351 # noqa: E501 - allowlist = ['aten::_convolution', 'aten::_convolution_nogroup', - 'aten::conv1d', 'aten::conv2d', 'aten::conv3d', 'aten::conv_tbc', + allowlist = ['aten::_convolution', 'aten::conv1d', 'aten::conv2d', 'aten::conv3d', 'aten::conv_tbc', 'aten::conv_transpose1d', 'aten::conv_transpose2d', 'aten::conv_transpose3d', 'aten::convolution', 'aten::cudnn_convolution', 'aten::cudnn_convolution_transpose', 'aten::prelu', 'aten::addmm', 'aten::addmv', 'aten::addr',