Hi @smuzaffar @akritkbehera
@kopperp has been looking at the recent versions of Eigen: they seem to have much better support for GPUs, and may not need our patches any more (see for example #10712).
However TensorFlow does not seem to like it (#10712 (comment)).
Any objections to try and update TF to the latest version currently 2.21 ?
Hi @smuzaffar @akritkbehera
@kopperp has been looking at the recent versions of Eigen: they seem to have much better support for GPUs, and may not need our patches any more (see for example #10712).
However TensorFlow does not seem to like it (#10712 (comment)).
Any objections to try and update TF to the latest version currently 2.21 ?