CMakeLists.txt 1.9 KB
Newer Older
W
Wu Yi 已提交
1 2
include(operators)

3
cc_library(reader_op_registry SRCS reader_op_registry.cc DEPS operator op_registry reader)
Y
Refine  
Yu Yang 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
set(LOCAL_READER_LIBS)

function(reader_library TARGET_NAME)
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    set(options "")
    set(common_deps reader_op_registry)
    cmake_parse_arguments(reader_library "${options}" "${oneValueArgs}"
            "${multiValueArgs}" ${ARGN})
    op_library(${TARGET_NAME} SRCS ${reader_library_SRCS} DEPS ${common_deps} ${reader_library_DEPS})
    set(LOCAL_READER_LIBS
            ${TARGET_NAME}
            ${LOCAL_READER_LIBS}
        PARENT_SCOPE)
endfunction()

Y
yuyang18 已提交
20
cc_library(buffered_reader SRCS buffered_reader.cc DEPS reader simple_threadpool)
Y
yuyang18 已提交
21
reader_library(open_files_op SRCS open_files_op.cc DEPS buffered_reader)
Y
Refine  
Yu Yang 已提交
22 23 24 25
reader_library(create_random_data_generator_op SRCS create_random_data_generator_op.cc)
reader_library(create_shuffle_reader_op SRCS create_shuffle_reader_op.cc)
reader_library(create_batch_reader_op SRCS create_batch_reader_op.cc)
reader_library(create_recordio_file_reader_op SRCS create_recordio_file_reader_op.cc)
Y
yuyang18 已提交
26
reader_library(create_double_buffer_reader_op SRCS create_double_buffer_reader_op.cc DEPS buffered_reader)
F
fengjiayi 已提交
27
reader_library(create_multi_pass_reader_op SRCS create_multi_pass_reader_op.cc)
28
reader_library(create_custom_reader_op SRCS create_custom_reader_op.cc)
S
sneaxiy 已提交
29
reader_library(create_py_reader_op SRCS create_py_reader_op.cc)
F
fengjiayi 已提交
30

Q
Qiao Longfei 已提交
31
if (NOT WIN32 AND NOT ON_INFER)
32 33 34
    cc_library(ctr_reader SRCS ctr_reader.cc DEPS gzstream reader zlib)
    cc_test(ctr_reader_test SRCS ctr_reader_test.cc DEPS ctr_reader)
    reader_library(create_ctr_reader_op SRCS create_ctr_reader_op.cc DEPS ctr_reader)
Q
Qiao Longfei 已提交
35
endif ()
36

F
fengjiayi 已提交
37
cc_test(reader_blocking_queue_test SRCS reader_blocking_queue_test.cc)
Y
Refine  
Yu Yang 已提交
38
# Export local libraries to parent
W
Wu Yi 已提交
39 40 41 42 43 44 45
# set(READER_LIBRARY ${LOCAL_READER_LIBS} PARENT_SCOPE)

op_library(read_op)

foreach(src ${LOCAL_READER_LIBS})
    set(OP_LIBRARY ${src} ${OP_LIBRARY} CACHE INTERNAL "op libs")
endforeach()