Hello!
Doing performance profiling wit nvprof for the PyTorch model with one convolution layer.
Setup:
For simplicity, the model consists of one Conv1d + Relu layer.
Input size: [32, 64, 1664] kernel_size: 11
Enabled cudnn.benchmark = True
Pre-warmed up.
In the profiler, I see that the volta_fp16_s884cudnn_fp16_256x128_ldg8_relu_f2f_exp_small_nhwc2nchw_tn_v1 function takes most of the time.
Could you tell me, please?
What is the purpose of this function?
Why are there no references to convolutions in the name of the function?
Link to json file with nvprof tracing info profile_one_layer_pytorch.sqlite.json · GitHub 1
File can be opened in chrome://tracing