[HLO Componentization] Create pass sub-component

PiperOrigin-RevId: 675222702
openxla · Sep 16, 2024 · d28ed3f · d28ed3f
1 parent 590cd6f
commit d28ed3f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xla/service/spmd/shardy/BUILD b/xla/service/spmd/shardy/BUILD
@@ -59,11 +59,11 @@ cc_library(
         "//xla:util",
         "//xla:xla_data_proto_cc",
         "//xla/hlo/ir:hlo",
+        "//xla/hlo/pass:hlo_pass",
         "//xla/hlo/utils:hlo_sharding_util",
         "//xla/mlir_hlo:mhlo_passes",
         "//xla/service:computation_layout",
         "//xla/service:hlo_dce",
-        "//xla/service:hlo_pass",
         "//xla/service:hlo_proto_cc",
         "//xla/service:tuple_simplifier",
         "//xla/service/llvm_ir:llvm_util",

diff --git a/xla/service/spmd/shardy/shardy_xla_pass.h b/xla/service/spmd/shardy/shardy_xla_pass.h
@@ -22,7 +22,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
-#include "xla/service/hlo_pass_interface.h"
+#include "xla/hlo/pass/hlo_pass_interface.h"
 
 namespace xla {
 namespace sdy {