问题分析:
torch.jit.fuser 的问题
解决方案:
torch._C._jit_override_can_fuse_on_cpu(False)
torch._C._jit_override_can_fuse_on_gpu(False)
torch._C._jit_set_texpr_fuser_enabled(False)
torch._C._jit_set_nvfuser_enabled(False)
torch.jit.fuser('off')
Reference:
pytorch discuss