Skip to content

Commit 5d45d10

Browse files
Varchocopybara-github
authored andcommitted
[SDY+ModelExplorer][1/4] Basic SDY support in model explorer. Register Shardy dialect in the mlir-to-json converter.
We will support SDY ops in the Model Explorer MLIR-JSON converter by 1) **(This CL)** Registering the SDY Dialect 2) Adding hierarchical node information to SDY ops 3) Display sharding attributes on nodes / ops 4) Introducing specific logic for ops with nested regions PiperOrigin-RevId: 744118602
1 parent 1ba08b6 commit 5d45d10

File tree

1 file changed

+4
-0
lines changed
  • shardy/dialect/sdy/ir

1 file changed

+4
-0
lines changed

shardy/dialect/sdy/ir/BUILD

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,10 @@ cc_library(
147147
"utils.h",
148148
"verifiers.h",
149149
],
150+
visibility = [
151+
"//shardy:friends",
152+
"//third_party/tensorflow/compiler/mlir/lite/experimental/google/tooling:__pkg__",
153+
],
150154
deps = [
151155
":attrs_inc",
152156
":bytecode_inc",

0 commit comments

Comments
 (0)