TensorFlow 和 OpenCL

TensorFlow and OpenCL

好像TensorFlow只支持CUDA,不支持OpenCL。

我看到了tensorflow-cl项目,它把CUDA代码编译成OpenCL,但它仍然是一个开发版本,并不是在所有情况下都有效。

我的问题是 TensorFlow 的开发者 Google 是否会开发其工具的多平台版本(不,我不是指 CPU 唯一版本)。专有 CUDA 的特性是否对专注于单一 GPU 供应商如此重要?有没有计划在未来的任何时候开发一个 OpenCL/Vulkan/SPIR-V 版本?

答案显然是肯定的,Tensorflow 在最近几周开始支持 OpenCL SYCL,仍处于 master github 分支,基本内核很少。许多其他内核需要以新格式编写,欢迎贡献。