Devices and external libraries options, make install, Makefile fix

sjperkins / tfopgen

Generate C++ and CUDA boilerplate for tensorflow custom operators

MIT License

20 stars 4 forks source link

Devices and external libraries options, make install, Makefile fix #4

Closed nicolaspi closed 2 years ago

nicolaspi commented 7 years ago

New features that allow to generate code according to operator's devices compatibility (cpu and/or gpu), and to specify external libraries to be linked.
The Makefile has been fixed to work with the latest tensorflow version (tested on the version 1.3 built from source).
Added 'make install' that copies the built library to the tensorflow's library directory.

nicolaspi commented 7 years ago

Looks like a version issue with tensorflow, the linker does not find libtensorflow_framework.so which was introduced by this commit (19 days ago).

sjperkins commented 7 years ago

Thanks for the PR. It looks like the new libtensorflow_framework.so linking style is going into tensorflow 1.4.0 (see https://github.com/tensorflow/tensorflow/issues/13607 for e.g.). It would probably be best to wait for this release before merging this and releasing a new tfopgen.