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

Add td_includes argument to allow more flexible relative include paths for td files.

Browse Source 
PiperOrigin-RevId: 268153974
This commit is contained in:
A. Unique TensorFlower 2019-09-09 22:40:26 -07:00 committed by TensorFlower Gardener
parent 191f36cb17
commit 14b473fc1b
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
third_party/mlir/tblgen.bzl vendored
View File

@ -1,6 +1,6 @@
"""BUILD extensions for MLIR table generation."""
def gentbl(name, tblgen, td_file, tbl_outs, td_srcs = [], strip_include_prefix = None):
def gentbl(name, tblgen, td_file, tbl_outs, td_srcs = [], td_includes = [], strip_include_prefix = None):
"""gentbl() generates tabular code from a table definition file.
Args:
@ -11,6 +11,7 @@ def gentbl(name, tblgen, td_file, tbl_outs, td_srcs = [], strip_include_prefix =
options passed to tblgen, and the out is the corresponding output file
produced.
td_srcs: A list of table definition files included transitively.
td_includes: A list of include paths for relative includes.
strip_include_prefix: attribute to pass through to cc_library.
"""
srcs = []
@ -20,8 +21,10 @@ def gentbl(name, tblgen, td_file, tbl_outs, td_srcs = [], strip_include_prefix =
# Add google_mlir/include directory as include so derived op td files can
# import relative to that.
td_includes = "-I external/local_config_mlir/include -I external/org_tensorflow "
td_includes += "-I $$(dirname $(location %s)) " % td_file
td_includes_str = "-I external/local_config_mlir/include -I external/org_tensorflow "
for td_include in td_includes:
td_includes_str += "-I %s " % td_include
td_includes_str += "-I $$(dirname $(location %s)) " % td_file
for (opts, out) in tbl_outs:
rule_suffix = "_".join(opts.replace("-", "_").replace("=", "_").split(" "))
native.genrule(
@ -32,7 +35,7 @@ def gentbl(name, tblgen, td_file, tbl_outs, td_srcs = [], strip_include_prefix =
message = "Generating code from table: %s" % td_file,
cmd = (("$(location %s) %s %s $(location %s) -o $@") % (
tblgen,
td_includes,
td_includes_str,
opts,
td_file,
)),