commit f590b3356d90f9e2b22b32ff6eeff193d7b585b9
author iree-copybara-bot <iree-copybara-bot@google.com> Tue Mar 17 10:13:15 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Tue Mar 17 10:14:36 2020 -0700
tree d8ec709948e19db73baf76a9decb2c277c8e481a
parent b8eb25c693e4f5a326b2cd7a6a7fcb1093ff96e3

Add Loop dialect to ModelBuilder.

PiperOrigin-RevId: 301398662

experimental/ModelBuilder/BUILD

diff --git a/experimental/ModelBuilder/BUILD b/experimental/ModelBuilder/BUILD
index 077bc52..11dab7f 100644
--- a/experimental/ModelBuilder/BUILD
+++ b/experimental/ModelBuilder/BUILD
@@ -30,6 +30,7 @@
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:LinalgOps",
         "@llvm-project//mlir:LinalgTransforms",
+        "@llvm-project//mlir:LoopOps",
         "@llvm-project//mlir:StandardOps",
         "@llvm-project//mlir:VectorOps",
     ],

experimental/ModelBuilder/ModelBuilder.h

diff --git a/experimental/ModelBuilder/ModelBuilder.h b/experimental/ModelBuilder/ModelBuilder.h
index c84769d..b69d7a9 100644
--- a/experimental/ModelBuilder/ModelBuilder.h
+++ b/experimental/ModelBuilder/ModelBuilder.h
@@ -47,6 +47,7 @@
 
 #include "mlir/Dialect/AffineOps/EDSC/Intrinsics.h"
 #include "mlir/Dialect/Linalg/EDSC/Intrinsics.h"
+#include "mlir/Dialect/LoopOps/EDSC/Builders.h"
 #include "mlir/Dialect/StandardOps/EDSC/Intrinsics.h"
 #include "mlir/Dialect/VectorOps/EDSC/Intrinsics.h"