commit 60bf7be41f581528bf90783b97880489699d7ba1
author Tatiana Shpeisman <shpeisman@google.com> Thu Jul 26 18:09:20 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:46:04 2019 -0700
tree 87f769ed4c0584e5274916f749fca4412e46b36b
parent 548cd7fc68fc5ff48d24316a90803270a00e1d6b

Implement MLValue, statement operands, operation statement operands and values. ML functions now have full support for expressing operations. Induction variables, function arguments and return values are still todo.

PiperOrigin-RevId: 206253643

lib/IR/Operation.cpp

diff --git a/lib/IR/Operation.cpp b/lib/IR/Operation.cpp
index 92a01ba..f2f9eb4 100644
--- a/lib/IR/Operation.cpp
+++ b/lib/IR/Operation.cpp
@@ -39,10 +39,7 @@
   if (auto *inst = dyn_cast<OperationInst>(this)) {
     return inst->getNumOperands();
   } else {
-    auto *stmt = cast<OperationStmt>(this);
-    (void)stmt;
-    // TODO: Add operands to OperationStmt.
-    return 0;
+    return cast<OperationStmt>(this)->getNumOperands();
   }
 }
 
@@ -51,9 +48,7 @@
     return inst->getOperand(idx);
   } else {
     auto *stmt = cast<OperationStmt>(this);
-    (void)stmt;
-    // TODO: Add operands to OperationStmt.
-    abort();
+    return stmt->getOperand(idx);
   }
 }
 
@@ -62,9 +57,7 @@
     inst->setOperand(idx, cast<CFGValue>(value));
   } else {
     auto *stmt = cast<OperationStmt>(this);
-    (void)stmt;
-    // TODO: Add operands to OperationStmt.
-    abort();
+    stmt->setOperand(idx, cast<MLValue>(value));
   }
 }
 
@@ -74,9 +67,7 @@
     return inst->getNumResults();
   } else {
     auto *stmt = cast<OperationStmt>(this);
-    (void)stmt;
-    // TODO: Add results to OperationStmt.
-    return 0;
+    return stmt->getNumResults();
   }
 }
 
@@ -86,9 +77,7 @@
     return inst->getResult(idx);
   } else {
     auto *stmt = cast<OperationStmt>(this);
-    (void)stmt;
-    // TODO: Add operands to OperationStmt.
-    abort();
+    return stmt->getResult(idx);
   }
 }