[singa] branch dev updated: bug fix: change use_cudnn to use_dnnl

[wangwei]

This is an automated email from the ASF dual-hosted git repository.

wangwei pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git


The following commit(s) were added to refs/heads/dev by this push:
     new 7b5033b  bug fix: change use_cudnn to use_dnnl
     new e3c3e4a  Merge pull request #610 from chrishkchris/fix_use_dnnl
7b5033b is described below

commit 7b5033be86b35826423ea85a83531f9d3b3c82e6
Author: Chris Yeung <chrishkch...@yahoo.com.hk>
AuthorDate: Thu Feb 27 11:18:49 2020 +0800

    bug fix: change use_cudnn to use_dnnl
---
 src/api/model_operation.i | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/api/model_operation.i b/src/api/model_operation.i
index 9438385..c10ac79 100755
--- a/src/api/model_operation.i
+++ b/src/api/model_operation.i
@@ -109,7 +109,7 @@ class PoolingHandle {
   bool is_max_pooling;
 };
 
-#if USE_CUDNN
+#if USE_DNNL
 Tensor CpuPoolingForward(const PoolingHandle &ph, const Tensor &x);
 Tensor CpuPoolingBackward(const PoolingHandle &ph, const Tensor &dy,
                               const Tensor& x, const Tensor& y);