There are always bubbles before executing cudaKernelLaunch, eg:
Here is the code:
W = Linear(edge_feature_in, hidden_channels * 3) * C.view(-1, 1)
what does pytorch do in this bubble? Can I eliminate this?
pytorch