Can you use the GPU QR factorization algorithm from cuSOLVER in tensorflow?

281 Views Asked by At

I can see it's referenced in https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf), but the tensorflow QR op at https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc isn't registered for GPU usage. Can I somehow make my own op using Geqrf, or is there maybe another way?

1

There are 1 best solutions below

0
On BEST ANSWER

This line in the source code you linked gives a hint about the current situation.

TODO(rmlarsen, volunteers): Implement the kernels below.