From 2e0c366a83417bdfe14de5b3a49067510a6897c5 Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Tue, 22 Dec 2020 13:08:16 -0600
Subject: [PATCH] Add generated function for CUDNN in DFGraph header
---
hpvm/include/SupportHPVM/DFGraph.h | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/hpvm/include/SupportHPVM/DFGraph.h b/hpvm/include/SupportHPVM/DFGraph.h
index fae04b7c8f..57a82eee7e 100644
--- a/hpvm/include/SupportHPVM/DFGraph.h
+++ b/hpvm/include/SupportHPVM/DFGraph.h
@@ -401,6 +401,14 @@ public:
GenFuncs.GPUGenFunc = F;
GenFuncInfo.gpu_hasCPUFunc = isCPUFunc;
break;
+ case hpvm::CUDNN_TARGET:
+ if (GenFuncs.GPUGenFunc != NULL) {
+ DEBUG(errs() << "Warning: Second generated GPU function for node "
+ << FuncPointer->getName() << "\n");
+ }
+ GenFuncs.CUDNNGenFunc = F;
+ GenFuncInfo.cudnn_hasCPUFunc = isCPUFunc;
+ break;
case hpvm::CPU_OR_GPU_TARGET:
assert(false && "A node function should be set with a tag specifying its \
type, not the node hint itself\n");
--
GitLab