File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -5,11 +5,11 @@ proto_library(framework_proto SRCS framework.proto)
5
5
cc_library (ddim SRCS ddim.cc DEPS eigen3 boost )
6
6
cc_test (ddim_test SRCS ddim_test.cc DEPS ddim )
7
7
nv_test (dim_test SRCS dim_test.cu DEPS ddim )
8
- cc_library (data_type SRCS data_type.cc DEPS framework_proto ddim )
8
+ cc_library (data_type SRCS data_type.cc DEPS framework_proto ddim device_context )
9
9
if (WITH_GPU )
10
- nv_library (tensor SRCS tensor.cc tensor_util.cu DEPS place memory device_context data_type )
10
+ nv_library (tensor SRCS tensor.cc tensor_util.cu DEPS place memory data_type )
11
11
else ()
12
- cc_library (tensor SRCS tensor.cc tensor_util.cc DEPS place memory device_context data_type )
12
+ cc_library (tensor SRCS tensor.cc tensor_util.cc DEPS place memory data_type )
13
13
endif ()
14
14
15
15
cc_test (tensor_test SRCS tensor_test.cc DEPS tensor )
You can’t perform that action at this time.
0 commit comments