Well so I have might have broken the gpu version for okrolearn but now it is fixed. https://github.com/Okerew/okrolearn