import iree in compiler workspace instead of tf workspace (#1049)

* import iree in compiler workspace instead of tf workspace

related changes: pai-disc/tensorflow#29

Author: wyzero

pytorch_blade/WORKSPACE

@@ -176,3 +176,7 @@ tf_workspace1()
 load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
 
 tf_workspace0()
+
+load("@org_disc_compiler//build_tools/bazel:workspace.bzl", "disc_compiler_workspace")
+
+disc_compiler_workspace()

tao_compiler/WORKSPACE

@@ -1,4 +1,4 @@
-workspace(name = "disc_compiler")
+workspace(name = "org_disc_compiler")
 
 local_repository(
     name = "org_tensorflow",
@@ -26,3 +26,9 @@ tf_workspace1()
 load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
 
 tf_workspace0()
+
+load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
+
+load("//build_tools/bazel:workspace.bzl", "disc_compiler_workspace")
+
+disc_compiler_workspace()

tao_compiler/build_tools/bazel/BUILD

@@ -0,0 +1,16 @@
+load("@org_tensorflow//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
+
+
+def workspace():
+    tf_http_archive(
+        name = "iree-dialects",
+        sha256 = "f6ed70146b85d27c25cb4653cfb879d7954be426516b1cac230cbf5bb8b041da",
+        strip_prefix = "iree-7cd0a8cb6e1027188faa75b9361c4b10aab4707c/llvm-external-projects/iree-dialects",
+        urls = tf_mirror_urls("https://github.com/iree-org/iree/archive/7cd0a8cb6e1027188faa75b9361c4b10aab4707c.zip"),
+        patch_file = ["@org_disc_compiler//third_party/iree:StructuredTransformOpsExt.cpp.patch"],
+    )
+
+# Alias so it can be loaded without assigning to a different symbol to prevent
+# shadowing previous loads and trigger a buildifier warning.
+disc_compiler_workspace = workspace
+

tao_compiler/build_tools/bazel/workspace.bzl

@@ -0,0 +1,25 @@
+diff --git a/lib/Dialect/LinalgTransform/IR/StructuredTransformOpsExt.cpp b/iree-dialects/lib/Dialect/LinalgTransform/IR/StructuredTransformOpsExt.cpp
+index 3a09c0297..e9f03a5c3 100644
+--- a/lib/Dialect/LinalgTransform/IR/StructuredTransformOpsExt.cpp
++++ b/lib/Dialect/LinalgTransform/IR/StructuredTransformOpsExt.cpp
+@@ -590,7 +590,6 @@ forgetUnnecessaryHandles(transform::TransformState &state,
+         return !handlesUsedAfterTransform[handle];
+       });
+       if (allHandlesUnused) {
+-        listener->removeMappings(payload);
+         seen.insert(payload);
+       }
+     }
+@@ -603,10 +602,11 @@ forgetUnnecessaryHandles(transform::TransformState &state,
+     for (Operation *payload : state.getPayloadOps(result)) {
+       if (!payload || seen.contains(payload))
+         continue;
+-      listener->removeMappings(payload);
+       seen.insert(payload);
+     }
+   }
++
++  for (Operation* payload : seen) listener->removeMappings(payload);
+ }
+
+ DiagnosedSilenceableFailure transform_ext::CanonicalizedSequenceOp::apply(