Skip to content

xjump/tensorflow-cl

 
 

Repository files navigation

Tensorflow-cl

Run Tensorflow on OpenCL™ 1.2 devices

Summary

This repo was created from the original Tensorflow repository at:

Please see the main repository for full Tensorflow documentation. This readme will only focus on the OpenCL porting aspects of Tensorflow.

Test results, on v0.14.0 wheel

test Intel HD5500, beignet 1.2.1 NVIDIA 940M, driver v367.57
unit tests (py.test -v) pass pass
linear_regression.py slow, but works slow, but works
logistic_regression.py ok ok
nearest_neighbor.py ok (accuracy 0.92) ok (accuracy 0.92)
multilayer_perceptron.py missing adam missing adam
recurrent_network.py missing adam missing adam
autoencoder.py missing rmsprop

Installation

pip install --upgrade tensorflow-0.11.0rc0-py3-none-any.whl

If you want, you can build from source

Testing

Setup

pip install -r tensorflow/stream_executor/cl/test/requirements.txt

Run

py.test -v

Design/architecture

Related projects

DNN Libraries

OpenCL middleware

  • CLBlast BLAS for OpenCL
  • cuda-on-cl Compile CUDA apps for OpenCL
  • EasyCL Handles running kernels, passing in arguments etc, on OpenCL

News

About

OpenCL 1.2 implementation for Tensorflow

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 45.8%
  • Python 41.4%
  • Jupyter Notebook 6.2%
  • TypeScript 2.3%
  • HTML 1.7%
  • Shell 0.9%
  • Other 1.7%