From 8ce5eab8ee859fb75533df89a104e41af534a249 Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Tue, 14 Jan 2020 13:18:36 -0600
Subject: [PATCH] Fix preffered targets getter imeplementation in VISCUtils
header
---
hpvm/include/SupportVISC/VISCUtils.h | 24 +++++++++++++-----------
1 file changed, 13 insertions(+), 11 deletions(-)
diff --git a/hpvm/include/SupportVISC/VISCUtils.h b/hpvm/include/SupportVISC/VISCUtils.h
index 9625846c39..f2c0d0d276 100644
--- a/hpvm/include/SupportVISC/VISCUtils.h
+++ b/hpvm/include/SupportVISC/VISCUtils.h
@@ -539,24 +539,26 @@ visc::Target getPreferredTarget(Function* F) {
DEBUG(errs() << "Finding preferred target for " << F->getName() << "\n");
Module* M = F->getParent();
- auto PrefTarget = [=](StringRef Name) {
+ auto FoundPrefTarget = [=](StringRef Name) {
NamedMDNode* HintNode = M->getOrInsertNamedMetadata(Name);
for(unsigned i = 0; i < HintNode->getNumOperands(); i++) {
MDNode* N = HintNode->getOperand(i);
Value* FHint = dyn_cast<ValueAsMetadata>(N->getOperand(0).get())->getValue();
if(F == FHint)
return true;
- }
+ }
+ return false;
};
-
- if(PrefTarget("visc_hint_gpu")) return visc::GPU_TARGET;
- if(PrefTarget("visc_hint_spir")) return visc::SPIR_TARGET;
- if(PrefTarget("visc_hint_cudnn")) return visc::CUDNN_TARGET;
- if(PrefTarget("visc_hint_promise")) return visc::PROMISE_TARGET;
- if(PrefTarget("visc_hint_cpu_gpu")) return visc::CPU_OR_GPU_TARGET;
- if(PrefTarget("visc_hint_cpu_spir")) return visc::CPU_OR_SPIR_TARGET;
-
- return visc::CPU_TARGET;
+
+ if(FoundPrefTarget("visc_hint_cpu")) return visc::CPU_TARGET;
+ if(FoundPrefTarget("visc_hint_gpu")) return visc::GPU_TARGET;
+ if(FoundPrefTarget("visc_hint_spir")) return visc::SPIR_TARGET;
+ if(FoundPrefTarget("visc_hint_cudnn")) return visc::CUDNN_TARGET;
+ if(FoundPrefTarget("visc_hint_promise")) return visc::PROMISE_TARGET;
+ if(FoundPrefTarget("visc_hint_cpu_gpu")) return visc::CPU_OR_GPU_TARGET;
+ if(FoundPrefTarget("visc_hint_cpu_spir")) return visc::CPU_OR_SPIR_TARGET;
+
+ return visc::None;
}
--
GitLab