experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Include .h.inc file into the header files of the wheel package

Browse Source 
While working on inlcuding some header files in compiler directory
from a installed tf-nightly wheel package, we noticed that the
generated header files from llvm (*.h.inc) in `tensorflow/compiler`
directories are not packaged into the wheel package.

As a result, it is not possible to include the header files in `tensorflow/compiler`
because they include `*.h.inc` as well.

For example, `mlir/tensorflow/ir/tf_ops.h` needs both `mlir/tensorflow/ir/tf_interfaces.h.inc`
and `mlir/tensorflow/ir/tf_ops.h.inc"

This PR add `.h.inc` into the packaged wheel file so that it is possible to use
header files in a locall install.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-04-26 00:12:22 +00:00
parent 634c902f4a
commit da93b879aa
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/tools/pip_package/setup.py
View File

@ -252,6 +252,7 @@ headers = (
list(find_files('*.h', 'tensorflow/c')) +
list(find_files('*.h', 'tensorflow/cc')) +
list(find_files('*.h', 'tensorflow/compiler')) +
list(find_files('*.h.inc', 'tensorflow/compiler')) +
list(find_files('*.h', 'tensorflow/core')) +
list(find_files('*.h', 'tensorflow/python')) +
list(find_files('*.h', 'tensorflow/stream_executor')) +