Errors on GPU

[lsgos]

Is this repo supposed to work with tensorflow-gpu? It runs fine on the CPU for me but on gpu I get an obscure tensorflow error at the end of an epoch

tensorflow.python.framework.errors_impl.InvalidArgumentError: Got info = 33345 for batch index 0, expected info = 0. Debug_info = heevd
	 [[{{node NG/SelfAdjointEigV2_15}}]

I think this happens inside a matrix inversion routine, so I suspect it could be an issue with precision on the gpu - is this repo meant to be GPU-able?