diff --git a/hpvm/include/SupportHPVM/DFGraph.h b/hpvm/include/SupportHPVM/DFGraph.h
index fae04b7c8f9024b6122ba761261aed65e9b052d4..57a82eee7ee8895e337e726fccbe0039d1a10dd8 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");