Solved by ptrblck in post #4 
              
              
                If you could split the operation into chunks of data, it’s possible. 
E.g. if you would like to sum a tensor, you could slice the original tensor, push this slice to the device, sum the values, and transfer the result back to the host. 
Once all slices are done, you could sum all temporary results. 
…