diff options
| author | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-07-02 19:17:14 +0300 |
|---|---|---|
| committer | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-07-02 19:17:14 +0300 |
| commit | 9bcb4ee5a30de27f9ac8976f06f1c5ef952f7930 (patch) | |
| tree | ab7c6ce05d4c8758566bd6dd41cb2119e0c69715 /src/graph/node_operators_unary.h | |
| parent | f85e63d77972b4214550509558a2306f3ddbc989 (diff) | |
Fix #262 PoolingWrapper dependency
Diffstat (limited to 'src/graph/node_operators_unary.h')
| -rw-r--r-- | src/graph/node_operators_unary.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 1e0c71e6..acc84c9c 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -7,7 +7,9 @@ #include "graph/node.h" #include "tensors/tensor_operators.h" -//#include "tensors/gpu/cudnn_wrappers.h" +#ifdef CUDNN +#include "tensors/gpu/cudnn_wrappers.h" +#endif namespace marian { @@ -1068,6 +1070,7 @@ struct ShiftNodeOp : public UnaryNodeOp { // Ptr<sparse::CSR> lf_; //}; +#ifdef CUDNN class PoolingOp : public UnaryNodeOp { public: PoolingOp(Expr x, @@ -1101,6 +1104,7 @@ public: protected: PoolingWrapper pooling_; }; +#endif class PoolingWithMaskingOp : public UnaryNodeOp { public: |