From ae600b3ef7cefb82c569444b1477873e4c2229f3 Mon Sep 17 00:00:00 2001 From: Marcin Junczys-Dowmunt Date: Thu, 15 Feb 2018 17:42:43 -0800 Subject: more clean up in backend --- src/graph/node_operators_unary.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/graph/node_operators_unary.h') diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 8390a0c2..a3f27fd2 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -1,10 +1,11 @@ #pragma once -#include "graph/backend_gpu.h" +#include "tensors/tensor.h" +#include "tensors/gpu/backend.h" + #include "graph/node.h" #include "kernels/sparse.h" #include "kernels/tensor_operators.h" -#include "tensors/tensor.h" #include "functional/functional.h" #include "kernels/cudnn_wrappers.h" -- cgit v1.2.3