more coda separation

2 files changed, 6 insertions, 5 deletions

diff --git a/src/graph/expression_operators.cpp b/src/graph/expression_operators.cpp
index be40a0d4..a1c9faa4 100644
--- a/src/graph/expression_operators.cpp
+++ b/src/graph/expression_operators.cpp
@@ -1,5 +1,4 @@
 #include "graph/expression_operators.h"
-//#include "kernels/sparse.h"
 #include "layers/constructors.h"
 
 #include "graph/node_operators.h"
@@ -333,6 +332,8 @@ Expr shift(Expr a, Shape shift) {
 //  return Expression<LexicalProbNodeOp>(logits, att, eps, lf);
 //}
 
+#ifdef CUDA_FOUND
+
 Expr avg_pooling(
     Expr x,
     int height,
@@ -410,4 +411,6 @@ Expr pooling_with_masking(Expr x, Expr mask, int width, bool isEven) {
   return Expression<PoolingWithMaskingOp>(x, mask, width, isEven);
 }
 
+#endif
+
 }
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h
index e857e790..0ca2c2a2 100644
--- a/src/graph/node_operators_unary.h
+++ b/src/graph/node_operators_unary.h
@@ -1,15 +1,13 @@
 #pragma once
 
 #include "tensors/tensor.h"
-#include "tensors/gpu/backend.h"
+#include "tensors/backend.h"
 
 #include "graph/node.h"
-//#include "kernels/sparse.h"
 #include "tensors/tensor_operators.h"
 #include "functional/functional.h"
 
-#include "tensors/gpu/cudnn_wrappers.h"
-
+//#include "tensors/gpu/cudnn_wrappers.h"
 
 namespace marian {