Thanks for all your help so far! I have managed to get my split-step integrator working on the GPU, but I seem to be encountering some performance issues. The split-step loop initially runs very fast, but then becomes 200x slower after about 20 passes through the loop. It then stays at this speed for the rest of the loop. I go into more detail about this in my stackoverflow
question. Is this caused by not being careful enough with reikna? Or is there some way to address this problem with reikna? Sorry that this question isn't directly related to reikna, but I thought you might be a great resource!