I am trying to use the correlation function from https://github.com/NVlabs/PWC-Net and/or https://github.com/lliuz/ARFlow. I have updated both to use the static method implementation required for 1.6.0 and replaced THCudaTensor/at::Tensor with torch::Tensor.
If I try to test with reasonably sized tensors (i.e. 16, 8, 128, 128), it fails with an illegal memory access during the backward methods (after resizing blobs), however it will work fine with smaller shaped tensors (i.e. 16, 64, 64, 64), even if the volume is greater.
I haven’t touched the CUDA kernels for either of them, and I guess they should be fine as they’re both published (and one being from nvidia)…
Both of them are previously used older versions of CUDA (8&9) and Torch (0.4 and 1.1), could this also be the issue?
Source for my changes can be found here.

  
      
      GitHub
  
  
    

5had3z/stereo-to-all

The aim of this project is to generate a pointcloud with motion vectors and semantic segmentation of stereoscopic image sequences. Targeting real-time deployment on an nVIDIA Xavier. - 5had3z/stere...


  
  
    
    
  
  


Cheers