OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Automated Code Change

Browse Source 
PiperOrigin-RevId: 821628227
This commit is contained in:
A. Unique TensorFlower 2025-10-20 06:50:35 -07:00 committed by TensorFlower Gardener
parent 789f9c8c92
commit f32acff204
15 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
third_party/xla/xla/hlo/experimental/auto_sharding/BUILD vendored
View File

@ -39,6 +39,7 @@ cc_library(
":auto_sharding_device_mesh",
":auto_sharding_iopddl",
":auto_sharding_option",
":auto_sharding_proto_cc",
":auto_sharding_solver",
":auto_sharding_strategy",
":auto_sharding_util",
@ -86,6 +87,7 @@ cc_library(
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/time",
"@com_google_absl//absl/types:span",
"@com_google_protobuf//:protobuf_lite",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:statusor",
@ -143,6 +145,7 @@ cc_library(
"@com_google_absl//absl/strings:string_view",
"@com_google_absl//absl/time",
"@com_google_ortools//ortools/linear_solver:linear_solver_cc_proto",
"@com_google_protobuf//:protobuf_lite",
"@local_tsl//tsl/platform:fingerprint",
"@local_tsl//tsl/platform:hash",
"@local_tsl//tsl/platform:protobuf",
@ -283,6 +286,7 @@ cc_library(
"//xla/service/spmd:spmd_partitioner",
"@com_google_absl//absl/container:btree",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/types:span",
@ -351,6 +355,7 @@ xla_cc_binary(
"//xla/hlo/analysis:alias_info",
"//xla/hlo/ir:hlo",
"//xla/tools:hlo_module_loader",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/status",
"@local_tsl//tsl/platform:platform_port",
],
@ -456,6 +461,7 @@ xla_cc_test(
"@com_google_absl//absl/status",
"@com_google_googletest//:gtest",
"@com_google_googletest//:gtest_main",
"@com_google_protobuf//:protobuf_lite",
"@local_tsl//tsl/platform",
"@local_tsl//tsl/platform:statusor",
] + if_google(["@com_google_ortools//ortools/linear_solver:linear_solver_scip"]),
@ -489,6 +495,7 @@ cc_test(
":iopddl_lib",
"//xla/tsl/platform:status_matchers",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:status_matchers",
"@com_google_absl//absl/time",
"@com_google_googletest//:gtest_main",
],

2
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc vendored
View File

@ -16,7 +16,6 @@ limitations under the License.
#include "xla/hlo/experimental/auto_sharding/auto_sharding.h"
#include <algorithm>
#include <climits>
#include <cstddef>
#include <cstdint>
#include <cstdlib>
@ -50,6 +49,7 @@ limitations under the License.
#include "absl/types/span.h"
#include "xla/hlo/analysis/alias_info.h"
#include "xla/hlo/analysis/hlo_alias_analysis.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding.pb.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_cost_graph.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_device_mesh.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_iopddl.h"

6
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_cpsat_for_request_test.cc vendored
View File

@ -10,20 +10,18 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "xla/hlo/experimental/auto_sharding/auto_sharding_solver.h"
#include <cstdint>
#include <string>
#include <utility>
#include <vector>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include "absl/container/flat_hash_set.h"
#include "absl/log/check.h"
#include "absl/status/status.h"
#include "google/protobuf/repeated_ptr_field.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding.pb.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_iopddl.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_solver.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_strategy.h"
#include "xla/hlo/experimental/auto_sharding/iopddl.h"
#include "xla/tsl/util/proto/proto_matchers.h"

2
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_dot_handler.cc vendored
View File

@ -35,6 +35,7 @@ limitations under the License.
#include "absl/strings/str_join.h"
#include "absl/strings/string_view.h"
#include "absl/types/span.h"
#include "google/protobuf/repeated_field.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_device_mesh.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_option.h"
@ -52,6 +53,7 @@ limitations under the License.
#include "xla/service/hlo_cost_analysis.h"
#include "xla/service/sharding_propagation.h"
#include "xla/shape.h"
#include "xla/xla_data.pb.h"
#include "tsl/platform/errors.h"
namespace xla {

3
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_impl.cc vendored
View File

@ -14,13 +14,10 @@ limitations under the License.
==============================================================================*/
#include <cstddef>
#include <cstdint>
#include <optional>
#include <string>
#include <utility>
#include <vector>
#include "absl/container/btree_set.h"
#include "absl/container/flat_hash_map.h"
#include "absl/status/statusor.h"
#include "absl/strings/string_view.h"

1
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_memory.cc vendored
View File

@ -27,6 +27,7 @@ limitations under the License.
#include "absl/container/btree_set.h"
#include "absl/container/flat_hash_set.h"
#include "absl/log/log.h"
#include "google/protobuf/repeated_field.h"
#include "tsl/platform/protobuf.h"
namespace xla {

1
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_memory.h vendored
View File

@ -24,6 +24,7 @@ limitations under the License.
#include "absl/container/btree_set.h"
#include "absl/container/flat_hash_set.h"
#include "google/protobuf/repeated_field.h"
#include "tsl/platform/protobuf.h"
namespace xla {

1
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_memory_test.cc vendored
View File

@ -23,6 +23,7 @@ limitations under the License.
#include <gtest/gtest.h>
#include "absl/container/btree_set.h"
#include "absl/container/flat_hash_set.h"
#include "google/protobuf/repeated_field.h"
namespace xla {
namespace spmd {

2
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_runner.cc vendored
View File

@ -14,9 +14,11 @@ limitations under the License.
==============================================================================*/
#include <iostream>
#include <memory>
#include <ostream>
#include <string>
#include "absl/log/check.h"
#include "absl/status/status.h"
#include "xla/hlo/analysis/alias_info.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding.h"

1
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_solver_test.cc vendored
View File

@ -21,6 +21,7 @@ limitations under the License.
#include <gtest/gtest.h>
#include "absl/container/flat_hash_set.h"
#include "absl/log/check.h"
#include "google/protobuf/repeated_ptr_field.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding.pb.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_iopddl.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_strategy.h"

2
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_strategy.cc vendored
View File

@ -16,7 +16,6 @@ limitations under the License.
#include "xla/hlo/experimental/auto_sharding/auto_sharding_strategy.h"
#include <algorithm>
#include <cmath>
#include <cstddef>
#include <cstdint>
#include <functional>
@ -30,7 +29,6 @@ limitations under the License.
#include "absl/algorithm/container.h"
#include "absl/container/flat_hash_map.h"
#include "absl/container/flat_hash_set.h"
#include "absl/container/inlined_vector.h"
#include "absl/functional/function_ref.h"
#include "absl/log/check.h"
#include "absl/log/log.h"

1
third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_util.cc vendored
View File

@ -19,6 +19,7 @@ limitations under the License.
#include <cstddef>
#include <cstdint>
#include <functional>
#include <initializer_list>
#include <memory>
#include <optional>
#include <queue>

1
third_party/xla/xla/hlo/experimental/auto_sharding/cluster_environment.cc vendored
View File

@ -27,6 +27,7 @@ limitations under the License.
#include "absl/container/btree_set.h"
#include "absl/container/flat_hash_map.h"
#include "absl/log/check.h"
#include "absl/log/log.h"
#include "absl/types/span.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_device_mesh.h"
#include "xla/hlo/experimental/auto_sharding/auto_sharding_strategy.h"

3
third_party/xla/xla/hlo/experimental/auto_sharding/iopddl_test.cc vendored
View File

@ -16,11 +16,10 @@ limitations under the License.
#include "xla/hlo/experimental/auto_sharding/iopddl.h"
#include <string>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include "absl/status/status.h"
#include "absl/status/status_matchers.h"
#include "absl/time/time.h"
#include "xla/hlo/experimental/auto_sharding/solver.h"
#include "xla/tsl/platform/status_matchers.h"

1
third_party/xla/xla/hlo/experimental/auto_sharding/metrics.cc vendored
View File

@ -16,7 +16,6 @@ limitations under the License.
#include "xla/hlo/experimental/auto_sharding/metrics.h"
#include <cstdint>
#include <string>
#include "xla/tsl/lib/monitoring/counter.h"