Pytorch1.5升级1.6 CPP代码修改(RuntimeError: Error compiling objects for extension)

Pytorch1.5升级1.6 CPP代码修改(RuntimeError: Error compiling objects for extension)

AT_CHECK和THCState_getCurrentStream(state) 用 TORCH_CHECK 和 at::cuda::getCurrentCUDAStream()替代。
参考https://github.com/hyangwinter/flownet3d_pytorch/issues/8#issuecomment-663925414

你可能感兴趣的:(c++)