Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

TensorFlow and OpenCL [closed]

It seems TensorFlow only supports CUDA and not OpenCL.

I saw the tensorflow-cl project, which compiles the CUDA code into OpenCL, but it is still a development version which does not work in all cases.

My question is whether Google, TensorFlow's developer, will ever develop a multi-platform version of its tool (no, I do not mean the CPU only version). Are the features of proprietary CUDA so critical to focus on a single GPU vendor? Are there any plans to develop an OpenCL/Vulkan/SPIR-V version at anytime in the future?

like image 655
Pietro Avatar asked May 07 '26 05:05

Pietro


1 Answers

The answer is obviously yes, and Tensorflow started supporting OpenCL SYCL in the last few weeks, still in the master github branch and with few basic kernels. Many other kernels need to be written in the new format and contributions are welcome.

like image 120
fabrizioM Avatar answered May 09 '26 05:05

fabrizioM



Donate For Us

If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!