commit ed229132f1c4ea2ba0644fc345d8279e47a00565
author Nicolas Vasilache <ntv@google.com> Mon Sep 21 19:30:42 2020
committer Nicolas Vasilache <ntv@google.com> Tue Sep 22 08:13:22 2020
tree ddd4c9f341131c15e1c2d33c8c7181cb5142e313
parent 0304c6da1006f6b472a1e5c1d8776a9f35c9439d

[mlir][Linalg] Uniformize linalg.generic with named ops.

This revision allows representing a reduction at the level of linalg on tensors for generic ops by uniformizing with the named ops approach.

mlir/lib/Conversion/LinalgToSPIRV/LinalgToSPIRV.cpp

diff --git a/mlir/lib/Conversion/LinalgToSPIRV/LinalgToSPIRV.cpp b/mlir/lib/Conversion/LinalgToSPIRV/LinalgToSPIRV.cpp
index 4c0aa75..18e2693 100644
--- a/mlir/lib/Conversion/LinalgToSPIRV/LinalgToSPIRV.cpp
+++ b/mlir/lib/Conversion/LinalgToSPIRV/LinalgToSPIRV.cpp
@@ -71,7 +71,7 @@
     return llvm::None;
 
   // Make sure this is reduction with one input and one output.
-  if (genericOp.args_in() != 1 || genericOp.args_out() != 1)
+  if (genericOp.getNumInputs() != 1 || genericOp.getNumOutputs() != 1)
     return llvm::None;
 
   auto originalInputType = op->getOperand(0).getType().cast<MemRefType>();