388f975cf5
* Added plugin with DPC++-based predictor and objective function * Update CMakeLists.txt * Update regression_obj_oneapi.cc * Added README.md for OneAPI plugin * Added OneAPI predictor support to gbtree * Update README.md * Merged kernels in gradient computation. Enabled multiple loss functions with DPC++ backend * Aligned plugin CMake files with latest master changes. Fixed whitespace typos * Removed debug output * [CI] Make oneapi_plugin a CMake target * Added tests for OneAPI plugin for predictor and obj. functions * Temporarily switched to default selector for device dispacthing in OneAPI plugin to enable execution in environments without gpus * Updated readme file. * Fixed USM usage in predictor * Removed workaround with explicit templated names for DPC++ kernels * Fixed warnings in plugin tests * Fix CMake build of gtest Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>
35 lines
1.4 KiB
CMake
35 lines
1.4 KiB
CMake
if (PLUGIN_LZ4)
|
|
target_sources(objxgboost PRIVATE ${xgboost_SOURCE_DIR}/plugin/lz4/sparse_page_lz4_format.cc)
|
|
target_link_libraries(objxgboost PUBLIC lz4)
|
|
endif (PLUGIN_LZ4)
|
|
|
|
if (PLUGIN_DENSE_PARSER)
|
|
target_sources(objxgboost PRIVATE ${xgboost_SOURCE_DIR}/plugin/dense_parser/dense_libsvm.cc)
|
|
endif (PLUGIN_DENSE_PARSER)
|
|
|
|
if (PLUGIN_UPDATER_ONEAPI)
|
|
add_library(oneapi_plugin OBJECT
|
|
${xgboost_SOURCE_DIR}/plugin/updater_oneapi/regression_obj_oneapi.cc
|
|
${xgboost_SOURCE_DIR}/plugin/updater_oneapi/predictor_oneapi.cc)
|
|
target_include_directories(oneapi_plugin
|
|
PRIVATE
|
|
${xgboost_SOURCE_DIR}/include
|
|
${xgboost_SOURCE_DIR}/dmlc-core/include
|
|
${xgboost_SOURCE_DIR}/rabit/include)
|
|
target_compile_definitions(oneapi_plugin PUBLIC -DXGBOOST_USE_ONEAPI=1)
|
|
target_link_libraries(oneapi_plugin PUBLIC -fsycl)
|
|
set_target_properties(oneapi_plugin PROPERTIES
|
|
COMPILE_FLAGS -fsycl
|
|
CXX_STANDARD 14
|
|
CXX_STANDARD_REQUIRED ON
|
|
POSITION_INDEPENDENT_CODE ON)
|
|
if (USE_OPENMP)
|
|
find_package(OpenMP REQUIRED)
|
|
target_link_libraries(oneapi_plugin PUBLIC OpenMP::OpenMP_CXX)
|
|
endif (USE_OPENMP)
|
|
# Get compilation and link flags of oneapi_plugin and propagate to objxgboost
|
|
target_link_libraries(objxgboost PUBLIC oneapi_plugin)
|
|
# Add all objects of oneapi_plugin to objxgboost
|
|
target_sources(objxgboost INTERFACE $<TARGET_OBJECTS:oneapi_plugin>)
|
|
endif (PLUGIN_UPDATER_ONEAPI)
|