Update TensorFlow submodule and fix python binding test
Closes https://github.com/google/iree/pull/2302
PiperOrigin-RevId: 318071207
diff --git a/SUBMODULE_VERSIONS b/SUBMODULE_VERSIONS
index 708ba96..111c821 100644
--- a/SUBMODULE_VERSIONS
+++ b/SUBMODULE_VERSIONS
@@ -11,7 +11,7 @@
b73f111094da3e380a1774b56b15f16c90ae8e23 third_party/sdl2
f8bf11a0253a32375c32cad92c841237b96696c0 third_party/spirv_headers
57eb48aed36160c4876bc8310d9ca84d42ee9e2a third_party/swiftshader
-080851ad09152127ec7664c626b30c71268009fb third_party/tensorflow
+d04bf998887fffe640ec8cacc9094574cd596f99 third_party/tensorflow
864d86e8b6d21449474db5e9313dbff90aa9c24f third_party/tracy
8a457f8552d8d47ce3a96ed80a714ff6396f8ad8 third_party/vulkan_extensionlayer
9bd3f561bcee3f01d22912de10bb07ce4e23d378 third_party/vulkan_headers
diff --git a/build_tools/bazel/third_party_import/llvm-project/overlay/llvm/BUILD.bazel b/build_tools/bazel/third_party_import/llvm-project/overlay/llvm/BUILD.bazel
index c70ff55..50ff746 100644
--- a/build_tools/bazel/third_party_import/llvm-project/overlay/llvm/BUILD.bazel
+++ b/build_tools/bazel/third_party_import/llvm-project/overlay/llvm/BUILD.bazel
@@ -561,6 +561,7 @@
name = "common_target_td_sources",
srcs = glob([
"include/llvm/CodeGen/*.td",
+ "include/llvm/Frontend/Directive/*.td",
"include/llvm/IR/Intrinsics*.td",
"include/llvm/TableGen/*.td",
"include/llvm/Target/*.td",
@@ -666,6 +667,17 @@
],
)
+gentbl(
+ name = "omp_gen",
+ tbl_outs = [("--gen-directive-decls", "include/llvm/Frontend/OpenMP/OMP.h.inc")],
+ tblgen = ":llvm-tblgen",
+ td_file = "include/llvm/Frontend/OpenMP/OMP.td",
+ td_srcs = glob([
+ "include/llvm/Frontend/OpenMP/*.td",
+ "include/llvm/Frontend/Directive/*.td",
+ ]),
+)
+
########################## Begin generated content ##########################
cc_library(
name = "AArch64AsmParser",
@@ -2053,6 +2065,7 @@
":Support",
":TransformUtils",
":config",
+ ":omp_gen",
],
)
diff --git a/build_tools/bazel/third_party_import/llvm-project/overlay/mlir/BUILD.bazel b/build_tools/bazel/third_party_import/llvm-project/overlay/mlir/BUILD.bazel
index 8fd0a94..ba9b580 100644
--- a/build_tools/bazel/third_party_import/llvm-project/overlay/mlir/BUILD.bazel
+++ b/build_tools/bazel/third_party_import/llvm-project/overlay/mlir/BUILD.bazel
@@ -1507,6 +1507,7 @@
":StandardToLLVM",
":Support",
":Transforms",
+ "@llvm-project//llvm:Support",
],
)
diff --git a/integrations/tensorflow/bindings/python/pyiree/tf/support/BUILD b/integrations/tensorflow/bindings/python/pyiree/tf/support/BUILD
index a25a8e4..c211cf0 100644
--- a/integrations/tensorflow/bindings/python/pyiree/tf/support/BUILD
+++ b/integrations/tensorflow/bindings/python/pyiree/tf/support/BUILD
@@ -16,6 +16,7 @@
"//bindings/python:build_defs.oss.bzl",
"INTREE_TENSORFLOW_PY_DEPS",
"iree_py_library",
+ "iree_py_test",
)
package(
@@ -36,7 +37,7 @@
],
)
-py_test(
+iree_py_test(
name = "tf_test_utils_test",
srcs = [
"tf_test_utils.py",
diff --git a/third_party/tensorflow b/third_party/tensorflow
index 080851a..d04bf99 160000
--- a/third_party/tensorflow
+++ b/third_party/tensorflow
@@ -1 +1 @@
-Subproject commit 080851ad09152127ec7664c626b30c71268009fb
+Subproject commit d04bf998887fffe640ec8cacc9094574cd596f99
