commit d8480cb7a231b229a4709350120d1838ed765cc3
author Thomas Raoux <thomasraoux@google.com> Fri Jun 19 18:26:30 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Jun 19 18:28:47 2020 -0700
tree 97a786d5b9df00576de1abb888d557b7a36f0749
parent 1316ad23852252df90f1241d6c8260d89c455ea3

Fix build of model builder test after interface change.

PiperOrigin-RevId: 317415534

experimental/ModelBuilder/test/TestSimpleJIT.cpp

diff --git a/experimental/ModelBuilder/test/TestSimpleJIT.cpp b/experimental/ModelBuilder/test/TestSimpleJIT.cpp
index bf0155b..9a2b29c 100644
--- a/experimental/ModelBuilder/test/TestSimpleJIT.cpp
+++ b/experimental/ModelBuilder/test/TestSimpleJIT.cpp
@@ -169,7 +169,7 @@
   OpBuilder b(&func.getBody());
   ScopedContext scope(b, func.getLoc());
   Value A(func.getArgument(0)), B(func.getArgument(1)), C(func.getArgument(2));
-  auto contractionBuilder = [](ArrayRef<BlockArgument> args) {
+  auto contractionBuilder = [](ValueRange args) {
     assert(args.size() == 3 && "expected 3 block arguments");
     (linalg_yield(vector_contraction_matmul(args[0], args[1], args[2])));
   };