Merge main -> google
* f8b466e2 Bumping tracy version and fixing API usage. (#3328)
* c2bd3b9d Refactor tf bindings compilation API (#3326)
* 9c698256 [vulkan] Create a GUI application to run an IREE module (#3274)
* 61cb3a6e Remove e2e test case decorator (#3304)
* e5953855 Canonicalizes mhlo.dot_general to a rank-3 mhlo.dot_general or mhlo.dot (#3225)
* f7cf2195 Merge google -> main (#3321)
* f25eef3c Update android benchmarking titles to match.. (#3322)
COPYBARA_INTEGRATE_REVIEW=https://github.com/google/iree/pull/3334 from hanhanW:main-to-google 62c8861c803837bfb3b67ac102b6afe4e4c299b0
PiperOrigin-RevId: 335072918
diff --git a/integrations/tensorflow/bindings/python/pyiree/tf/compiler/signature_def_saved_model_test.py b/integrations/tensorflow/bindings/python/pyiree/tf/compiler/signature_def_saved_model_test.py
index 2640862..9764b13 100644
--- a/integrations/tensorflow/bindings/python/pyiree/tf/compiler/signature_def_saved_model_test.py
+++ b/integrations/tensorflow/bindings/python/pyiree/tf/compiler/signature_def_saved_model_test.py
@@ -56,7 +56,7 @@
builder.save()
module = compiler.tf_signature_def_saved_model_to_compiler_module(
- sm_dir, tags=set(["bar"]), exported_names=["baz"])
+ sm_dir, saved_model_tags=set(["bar"]), exported_names=["baz"])
module_asm = module.to_asm(large_element_limit=100)
self.assertRegexpMatches(module_asm, "flow.variable @[^ ]* dense<10>")
diff --git a/iree/compiler/Dialect/Flow/Transforms/test/hlo_to_hlo_preprocessing_canoncalize_dot_general.mlir b/iree/compiler/Dialect/Flow/Transforms/test/hlo_to_hlo_preprocessing_canoncalize_dot_general.mlir
index 3447436..67b8f46 100644
--- a/iree/compiler/Dialect/Flow/Transforms/test/hlo_to_hlo_preprocessing_canoncalize_dot_general.mlir
+++ b/iree/compiler/Dialect/Flow/Transforms/test/hlo_to_hlo_preprocessing_canoncalize_dot_general.mlir
@@ -15,7 +15,7 @@
// CHECK: dot_general_to_dot(%[[ARG0:.+]]: tensor<1x32x128x4xf32>, %[[ARG1:.+]]: tensor<128x4x8x64xf32>) -> tensor<1x32x8x64xf32>
// CHECK: %[[ARG0_RESHAPED:.+]] = "mhlo.reshape"(%[[ARG0]]) : (tensor<1x32x128x4xf32>) -> tensor<32x512xf32>
// CHECK: %[[ARG1_RESHAPED:.+]] = "mhlo.reshape"(%[[ARG1]]) : (tensor<128x4x8x64xf32>) -> tensor<512x512xf32>
-// CHECK: %[[DOT:.+]] = "mhlo.dot"(%[[ARG0_RESHAPED]], %[[ARG1_RESHAPED]])
+// CHECK: %[[DOT:.+]] = "mhlo.dot"(%[[ARG0_RESHAPED]], %[[ARG1_RESHAPED]])
// CHECK: %[[RESULT:.+]] = "mhlo.reshape"(%[[DOT]]) : (tensor<32x512xf32>) -> tensor<1x32x8x64xf32>
// CHECK: return %[[RESULT]] : tensor<1x32x8x64xf32>
