> > Le 18 oct. 2018 à 00:15, Wong Hang <[email protected]> a écrit : > > Hi there, > > May I ask why theano gpuarray does not support float64 cholesky factorization > and triangular solve? > Are there any reason behind that?
The main reason is that there is little interest in float64 because it is *much* slower than float32 on GPUs. > I think I can simply hack the code in theano/gpuarray/linalg.py to invoke > cusolver.cusolverDnDpotrf / cublas.cublasDtrsm / cublas.cublasDtrsv rather > than cusolver.cusolverDnSpotrf / cublas.cublasStrsm / cublas.cublasStrsv to > get float64 support? > Yes, that should work, but make sure to update the surrounding support code so that it checks for float64 rather than float32. You could also make it switch the function used based on the input argument types. > Best, > Hang > > -- > > --- > You received this message because you are subscribed to the Google Groups > "theano-users" group. > To unsubscribe from this group and stop receiving emails from it, send an > email to [email protected] > <mailto:[email protected]>. > For more options, visit https://groups.google.com/d/optout > <https://groups.google.com/d/optout>. -- --- You received this message because you are subscribed to the Google Groups "theano-users" group. To unsubscribe from this group and stop receiving emails from it, send an email to [email protected]. For more options, visit https://groups.google.com/d/optout.
