Automated Code Change

PiperOrigin-RevId: 675843073
openxla · Sep 18, 2024 · c7c6d44 · c7c6d44
1 parent 10dded8
commit c7c6d44
Show file tree

Hide file tree

Showing 13 changed files with 40 additions and 6 deletions.
diff --git a/xla/translate/mhlo_to_hlo/BUILD b/xla/translate/mhlo_to_hlo/BUILD
@@ -29,6 +29,7 @@ cc_library(
         "//xla/service:hlo_proto_cc",
         "//xla/stream_executor:dnn",
         "@com_google_absl//absl/status:statusor",
+        "@com_google_absl//absl/types:span",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Support",
@@ -47,6 +48,8 @@ cc_library(
         "//xla/hlo/ir:hlo",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
+        "@tsl//tsl/platform:errors",
+        "@tsl//tsl/platform:statusor",
     ],
 )
 
@@ -57,7 +60,6 @@ cc_library(
     deps = [
         ":stack_frame_index_builder",
         "//xla:xla_data_proto_cc",
-        "@com_google_absl//absl/log",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Support",
@@ -71,6 +73,7 @@ cc_library(
     deps = [
         "//xla:xla_data_proto_cc",
         "//xla/service:hlo_module_config",
+        "//xla/service:hlo_proto_cc",
         "@com_google_absl//absl/status",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
@@ -85,6 +88,7 @@ cc_library(
     deps = [
         "//xla/service:hlo_proto_cc",
         "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Support",
     ],
 )
 
@@ -280,7 +284,10 @@ xla_cc_test(
         "//xla:shape_util",
         "//xla:test",
         "//xla:xla_data_proto_cc",
+        "//xla/mlir_hlo",
         "//xla/translate/hlo_to_mhlo:hlo_utils",
+        "@com_google_absl//absl/status:statusor",
+        "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
         "@tsl//tsl/platform:protobuf",
         "@tsl//tsl/platform:test_main",

diff --git a/xla/translate/mhlo_to_hlo/attribute_exporter.cc b/xla/translate/mhlo_to_hlo/attribute_exporter.cc
@@ -18,16 +18,17 @@ limitations under the License.
 #include <utility>
 
 #include "absl/status/statusor.h"
+#include "absl/types/span.h"
 #include "llvm/Support/LogicalResult.h"
 #include "mlir/IR/Attributes.h"
 #include "mlir/IR/BuiltinAttributes.h"
 #include "mlir/Support/LLVM.h"
 #include "stablehlo/dialect/Base.h"
+#include "xla/hlo/ir/hlo_sharding.h"
 #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/service/hlo_parser.h"
 #include "xla/shape_util.h"
-#include "xla/stream_executor/dnn.h"
-#include "xla/types.h"
 #include "xla/util.h"
 #include "xla/xla_data.pb.h"
 

diff --git a/xla/translate/mhlo_to_hlo/attribute_exporter.h b/xla/translate/mhlo_to_hlo/attribute_exporter.h
@@ -19,7 +19,11 @@ limitations under the License.
 #include <utility>
 
 #include "absl/status/statusor.h"
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/ADT/StringRef.h"
 #include "mlir/IR/Attributes.h"
+#include "mlir/IR/BuiltinAttributes.h"
+#include "mlir/Support/LLVM.h"
 #include "xla/hlo/ir/hlo_input_output_alias_config.h"
 #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"
 #include "xla/service/hlo.pb.h"

diff --git a/xla/translate/mhlo_to_hlo/layout_util.cc b/xla/translate/mhlo_to_hlo/layout_util.cc
@@ -16,6 +16,14 @@ limitations under the License.
 #include "xla/translate/mhlo_to_hlo/layout_util.h"
 
 #include "absl/status/status.h"
+#include "absl/status/statusor.h"
+#include "xla/client/xla_builder.h"
+#include "xla/hlo/ir/hlo_sharding.h"
+#include "xla/shape.h"
+#include "xla/shape_util.h"
+#include "xla/xla_data.pb.h"
+#include "tsl/platform/errors.h"
+#include "tsl/platform/statusor.h"
 
 namespace mlir {
 

diff --git a/xla/translate/mhlo_to_hlo/location_exporter.cc b/xla/translate/mhlo_to_hlo/location_exporter.cc
@@ -17,7 +17,6 @@ limitations under the License.
 
 #include <string>
 
-#include "absl/log/log.h"
 #include "llvm/ADT/STLExtras.h"
 #include "llvm/ADT/SmallVector.h"
 #include "llvm/ADT/StringExtras.h"
@@ -28,6 +27,7 @@ limitations under the License.
 #include "mlir/IR/Visitors.h"
 #include "mlir/Support/LLVM.h"
 #include "xla/translate/mhlo_to_hlo/stack_frame_index_builder.h"
+#include "xla/xla_data.pb.h"
 
 namespace mlir {
 namespace mhlo {

diff --git a/xla/translate/mhlo_to_hlo/mlir_hlo_to_hlo.h b/xla/translate/mhlo_to_hlo/mlir_hlo_to_hlo.h
@@ -19,11 +19,15 @@ limitations under the License.
 #include <optional>
 #include <vector>
 
+#include "absl/status/status.h"
 #include "absl/status/statusor.h"
+#include "llvm/ADT/ArrayRef.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/IR/Block.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "xla/client/xla_builder.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/service/hlo_module_config.h"
 #include "xla/translate/mhlo_to_hlo/layout_util.h"
 

diff --git a/xla/translate/mhlo_to_hlo/module_attributes_exporter.cc b/xla/translate/mhlo_to_hlo/module_attributes_exporter.cc
@@ -25,6 +25,7 @@ limitations under the License.
 #include "mlir/IR/BuiltinAttributes.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/Support/LLVM.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/service/hlo_module_config.h"
 #include "xla/xla_data.pb.h"
 

diff --git a/xla/translate/mhlo_to_hlo/module_attributes_exporter.h b/xla/translate/mhlo_to_hlo/module_attributes_exporter.h
@@ -19,6 +19,7 @@ limitations under the License.
 #include "absl/status/status.h"
 #include "mlir/IR/BuiltinAttributes.h"
 #include "mlir/IR/BuiltinOps.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/service/hlo_module_config.h"
 
 namespace mlir {

diff --git a/xla/translate/mhlo_to_hlo/operator_writer_gen.cc b/xla/translate/mhlo_to_hlo/operator_writer_gen.cc
@@ -20,13 +20,14 @@ limitations under the License.
 #include "llvm/ADT/StringExtras.h"
 #include "llvm/ADT/StringMap.h"
 #include "llvm/Support/CommandLine.h"
-#include "llvm/Support/FormatVariadic.h"
 #include "llvm/Support/InitLLVM.h"
 #include "llvm/Support/Signals.h"
 #include "llvm/TableGen/Error.h"
 #include "llvm/TableGen/Main.h"
 #include "llvm/TableGen/Record.h"
 #include "llvm/TableGen/TableGenBackend.h"
+#include "mlir/TableGen/Argument.h"
+#include "mlir/TableGen/Attribute.h"
 #include "mlir/TableGen/Operator.h"
 
 using llvm::interleaveComma;

diff --git a/xla/translate/mhlo_to_hlo/stack_frame_index_builder.cc b/xla/translate/mhlo_to_hlo/stack_frame_index_builder.cc
@@ -24,6 +24,7 @@ limitations under the License.
 
 #include "mlir/IR/BuiltinAttributes.h"
 #include "mlir/IR/Location.h"
+#include "mlir/Support/LLVM.h"
 #include "xla/service/hlo.pb.h"
 
 namespace mlir {

diff --git a/xla/translate/mhlo_to_hlo/translate.h b/xla/translate/mhlo_to_hlo/translate.h
@@ -21,6 +21,7 @@ limitations under the License.
 
 #include "llvm/Support/MemoryBuffer.h"
 #include "llvm/Support/raw_os_ostream.h"
+#include "llvm/Support/raw_ostream.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/Support/LogicalResult.h"
 

diff --git a/xla/translate/mhlo_to_hlo/type_to_shape.cc b/xla/translate/mhlo_to_hlo/type_to_shape.cc
@@ -26,6 +26,7 @@ limitations under the License.
 #include "llvm/ADT/ArrayRef.h"
 #include "llvm/ADT/STLExtras.h"
 #include "llvm/ADT/SmallVector.h"
+#include "llvm/Support/LogicalResult.h"
 #include "mlir/Dialect/SparseTensor/IR/Enums.h"
 #include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
 #include "mlir/IR/AffineMap.h"
@@ -35,7 +36,6 @@ limitations under the License.
 #include "mlir/IR/Location.h"
 #include "mlir/Support/DebugStringHelper.h"
 #include "mlir/Support/LLVM.h"
-#include "mlir/Support/LogicalResult.h"
 #include "stablehlo/dialect/StablehloOps.h"
 #include "xla/mlir/utils/type_util.h"
 #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"

diff --git a/xla/translate/mhlo_to_hlo/type_to_shape_test.cc b/xla/translate/mhlo_to_hlo/type_to_shape_test.cc
@@ -18,9 +18,14 @@ limitations under the License.
 #include <iostream>
 #include <utility>
 
+#include "absl/status/statusor.h"
+#include "llvm/ADT/SmallVector.h"
 #include "mlir/IR/Builders.h"
+#include "mlir/IR/BuiltinTypeInterfaces.h"
 #include "mlir/IR/BuiltinTypes.h"
 #include "mlir/IR/MLIRContext.h"
+#include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"
+#include "xla/shape.h"
 #include "xla/shape_util.h"
 #include "xla/test.h"
 #include "xla/translate/hlo_to_mhlo/hlo_utils.h"