Merge pull request #3185 from ROCm/develop-upstream-sync-260306

Develop upstream sync 260306

Author: nurmukhametov

.bazelrc

@@ -100,6 +100,7 @@ common --action_env=PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=upb
 # conflict when precompiling. Disable to avoid that problem.
 # See https://github.com/bazel-contrib/rules_python/issues/2445
 common --@rules_python//python/config_settings:precompile=force_disabled
+common --@rules_python//python/config_settings:bootstrap_impl=script --repo_env=RULES_PYTHON_ENABLE_PIPSTAR=0
 
 # Do not do this. If enabled protobuf's core internal target 
 # @com_google_protobuf//python:protobuf_python will start depending on a bunch

ci/devinfra/docker/windows2022/Dockerfile

@@ -4,7 +4,7 @@
 # - MSVC 14.39
 # - LLVM/Clang 18.1.4
 # - MSYS2 + curl, git, patch, vim, unzip, zip
-# - Python 3.9 - 3.13
+# - Python 3.9 - 3.14
 # - Bazelisk 1.22.1
 # - JDK 21 (Azul Zulu)
 
@@ -108,7 +108,8 @@ RUN function Install-Python { \
         @{ version = '3.10.11'; prependPath = 0 }, \
         @{ version = '3.11.9'; prependPath = 0 }, \
         @{ version = '3.12.8'; prependPath = 0 }, \
-        @{ version = '3.13.1'; prependPath = 1 } \
+        @{ version = '3.13.1'; prependPath = 0 } \
+        @{ version = '3.14.3'; prependPath = 1 } \
     ); \
     foreach ($v in $versions) { \
         Install-Python -version $v.version -prependPath $v.prependPath; \

tensorflow/compiler/jit/device_compiler_test.cc

@@ -480,7 +480,7 @@ TEST_F(DeviceCompilerTest, CompileStrictPersistentCacheFailedToPersist) {
   const XlaCompiler::CompilationResult* compilation_result = nullptr;
   xla::LocalExecutable* xla_executable = nullptr;
 
-  auto persistor = down_cast<MockXlaDeviceExecutablePersistor*>(
+  auto persistor = absl::down_cast<MockXlaDeviceExecutablePersistor*>(
       xla_device_compiler->persistor());
   TF_ASSERT_OK_AND_ASSIGN(auto signature, Signature::Build(fn, args));
   EXPECT_CALL(*persistor,

tensorflow/compiler/mlir/lite/BUILD

@@ -13,7 +13,7 @@ load(
 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
 
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],
 )
@@ -590,6 +590,7 @@ cc_library(
         ":cleanup_optimization_barrier",
         ":converter_inc",
         ":cost_estimators",
+        ":downcast_x64",
         ":elements_attr_roundtrip_pass",
         ":optimize_broadcast_like_pass",
         ":optimize_broadcast_like_pass_options",
@@ -1018,6 +1019,26 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "downcast_x64",
+    srcs = [
+        "transforms/downcast_x64_pass.cc",
+    ],
+    hdrs = [
+        "transforms/downcast_x64_pass.h",
+    ],
+    deps = [
+        ":pass",
+        ":pass_options",
+        ":tensorflow_lite_ops",
+        "@llvm-project//llvm:Support",
+        "@llvm-project//mlir:FuncDialect",
+        "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Pass",
+        "@llvm-project//mlir:Support",
+    ],
+)
+
 cc_library(
     name = "tensorflow_lite_legalize_tf_analyze_variables",
     srcs = [
@@ -1165,6 +1186,7 @@ cc_library(
         ":constant_utils",
         ":convert_type",
         ":cost_estimators",
+        ":downcast_x64",
         ":elements_attr_roundtrip_pass",
         ":fake_quant_utils",
         ":lstm_utils",
@@ -1433,6 +1455,7 @@ cc_library(
     ],
     deps = [
         "convert_type",
+        ":downcast_x64",
         ":elements_attr_roundtrip_pass",
         ":op_quant_spec_getters_inc",
         ":optimize_broadcast_like_pass_options",
@@ -1991,6 +2014,7 @@ cc_library(
         ":common",
         ":converter_flags_proto_cc",
         ":converter_pass_options_setter",
+        ":downcast_x64",
         ":elements_attr_roundtrip_pass",
         ":fake_quant_utils",
         ":optimize_broadcast_like_pass",

tensorflow/compiler/mlir/lite/converter_flags.proto

@@ -39,7 +39,7 @@ enum FileFormat {
 // of as properties of models, instead describing how models are to be
 // processed in the context of the present tooling job.
 //
-// Next ID to use: 69.
+// Next ID to use: 70.
 message ConverterFlags {
   reserved 54, 61;
 
@@ -380,4 +380,7 @@ message ConverterFlags {
   // but it may cause incorrect results when broadcasting ops are introduced by
   // explicit broadcasting in the source model.
   optional bool unsafe_fuse_dynamic_shaped_broadcast = 68 [default = false];
+
+  // If false, downcast x64 tensors and inputs to x32.
+  optional bool enable_x64 = 69 [default = true];
 }

tensorflow/compiler/mlir/lite/core/BUILD

@@ -3,7 +3,7 @@ load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
 load("//tensorflow/compiler/mlir/lite:build_def.bzl", "tflite_copts_warnings")
 
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
     licenses = ["notice"],
 )
 

tensorflow/compiler/mlir/lite/core/api/BUILD

@@ -4,7 +4,7 @@ load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
 load("//tensorflow/compiler/mlir/lite:build_def.bzl", "tflite_copts")
 
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],
 )

tensorflow/compiler/mlir/lite/core/c/BUILD

@@ -3,7 +3,7 @@ load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
 load("//tensorflow/compiler/mlir/lite:build_def.bzl", "tflite_copts")
 
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
     default_visibility = [
         "//visibility:public",
     ],

tensorflow/compiler/mlir/lite/delegates/BUILD

@@ -14,7 +14,7 @@
 # ==============================================================================
 
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],
 )

tensorflow/compiler/mlir/lite/delegates/flex/BUILD

@@ -18,7 +18,7 @@ default_visibility = [
 # This is a TF Lite delegate that is powered by TensorFlow's Eager.
 #
 package(
-    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
     default_visibility = default_visibility,
     licenses = ["notice"],
 )