diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-11-02 01:28:08 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-11-02 01:28:08 +0300 |
| commit | cd6fca847dd9aa0b5236306beaaa8f778e2d8726 (patch) | |
| tree | 495608b6ad01298da0b6e5478c3c6505e0cc2841 /src | |
| parent | 289c25cb2ca0df9077480f8e843e99ea2ef8f018 (diff) | |
commented out offending code make it compile, needs to be fixed when USE_CUDNN=on
Diffstat (limited to 'src')
| -rw-r--r-- | src/graph/node_operators_binary.h | 10 | ||||
| -rw-r--r-- | src/graph/node_operators_unary.h | 4 |
2 files changed, 8 insertions, 6 deletions
diff --git a/src/graph/node_operators_binary.h b/src/graph/node_operators_binary.h index 397f85db..bbdd4cee 100644 --- a/src/graph/node_operators_binary.h +++ b/src/graph/node_operators_binary.h @@ -775,13 +775,15 @@ public: nodes[2]->shape()[2], nodes[2]->shape()[3])); + /* @TODO: does not compile CUDNN_CALL(cudnnGetConvolution2dForwardOutputDim(convDesc_, xDesc_, kernelDesc_, - shape_.begin(), - shape_.begin() + 1, - shape_.begin() + 2, - shape_.begin() + 3)); + shape_.data(), + shape_.data() + 1, + shape_.data() + 2, + shape_.data() + 3)); + */ CUDNN_CALL(cudnnCreateTensorDescriptor(&yDesc_)); CUDNN_CALL(cudnnSetTensor4dDescriptor(yDesc_, diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 844f8905..fcf1c290 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -1047,14 +1047,14 @@ public: padWidth, strideHeight, strideWidth)); - + /* @TODO: does not compile CUDNN_CALL(cudnnGetPooling2dForwardOutputDim(poolingDesc_, xDesc_, shape_.begin(), shape_.begin() + 1, shape_.begin() + 2, shape_.begin() + 3)); - +*/ CUDNN_CALL(cudnnCreateTensorDescriptor(&yDesc_)); CUDNN_CALL(cudnnSetTensor4dDescriptor(yDesc_, CUDNN_TENSOR_NCHW, |