From f6227e165c5297efe01802e2ffe1a5a8d2810111 Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Mon, 21 Dec 2020 13:11:53 -0600
Subject: [PATCH] Move GenVISC to GenHPVM
---
hpvm/lib/Transforms/GenHPVM/GenHPVM.cpp | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/hpvm/lib/Transforms/GenHPVM/GenHPVM.cpp b/hpvm/lib/Transforms/GenHPVM/GenHPVM.cpp
index 2d62ffda0e..00063d6dc8 100644
--- a/hpvm/lib/Transforms/GenHPVM/GenHPVM.cpp
+++ b/hpvm/lib/Transforms/GenHPVM/GenHPVM.cpp
@@ -184,6 +184,8 @@ IS_HPVM_CALL(tensor_tanh)
IS_HPVM_CALL(tensor_sigmoid)
IS_HPVM_CALL(tensor_softmax)
+IS_HPVM_CALL(node_id)
+
// Return the constant integer represented by value V
static unsigned getNumericValue(Value *V) {
assert(isa<ConstantInt>(V) &&
@@ -762,6 +764,11 @@ bool GenHPVM::runOnModule(Module &M) {
if (isHPVMCall_tensor_softmax(I)) {
ReplaceCallWithIntrinsic(I, Intrinsic::hpvm_tensor_softmax, &toBeErased);
}
+
+ // New Intrinsic to set Node ID
+ if (isHPVMCall_node_id(I)) {
+ ReplaceCallWithIntrinsic(I, Intrinsic::hpvm_node_id, &toBeErased);
+ }
}
// Erase the __hpvm__node calls
--
GitLab