Solved by ptrblck in post #2 
              
              
                The workaround would be to use the CPU implementation, i.e. push the data to the CPU before applying this method and back to the GPU afterwards for now assuming that you need the deterministic outputs for debugging (as it would slow down your code).