Solved by glaringlee in post #2 
              
              
                @Separius 
CUDALoops.cuh is not exposed. 
TensorIterator is exposed. To use it, you can do 
#include <ATen/native/TensorIterator.h> 
For elementwise cuda kernel block setup, you can take a look at here, we have basic thread and block setup here. 
 
 
And within CUDALoops.cuh, you can check gpu_kerne…