Add the torch/nn/parallel/comm.py (from torch/cuda/comm.py) to white list.    The required attribute _broadcast, _broadcast_coalesced, _scatter and _gather are statically registered by the device runtime.