CMakeLists.txt

if (NOT LITE_WITH_OPENCL)
    return ()
endif()

set(cl_kernel_deps op_params_lite cl_caller cl_engine cl_context cl_wrapper)

cc_library(elementwise_add_opencl SRCS elementwise_add_compute.cc DEPS ${cl_kernel_deps})
cc_library(pool_opencl SRCS pool_compute.cc DEPS ${cl_kernel_deps})

lite_cc_test(test_elementwise_add_opencl SRCS elementwise_add_compute_test.cc DEPS elementwise_add_opencl
  op_registry_lite program_lite
  context_lite
  )

lite_cc_test(test_pool_opencl SRCS pool_compute_test.cc DEPS pool_opencl
  op_registry_lite program_lite
  context_lite
  )

set(opencl_kernels
  elementwise_add_opencl
  pool_opencl
  CACHE INTERNAL "opencl_kernels")