mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-17 15:06:37 +00:00
36d4e4c9b5
This patch implements the "task record and replay" mechanism. The idea is to be able to store tasks and their dependencies in the runtime so that we do not pay the cost of task creation and dependency resolution for future executions. The objective is to improve fine-grained task performance, both for those from "omp task" and "taskloop". The entry point of the recording phase is __kmpc_start_record_task, and the end of record is triggered by __kmpc_end_record_task. Tasks encapsulated between a record start and a record end are saved, meaning that the runtime stores their dependencies and structures, referred to as TDG, in order to replay them in subsequent executions. In these TDG replays, we start the execution by scheduling all root tasks (tasks that do not have input dependencies), and there will be no involvement of a hash table to track the dependencies, yet tasks do not need to be created again. At the beginning of __kmpc_start_record_task, we must check if a TDG has already been recorded. If yes, the function returns 0 and starts to replay the TDG by calling __kmp_exec_tdg; if not, we start to record, and the function returns 1. An integer uniquely identifies TDGs. Currently, this identifier needs to be incremented manually in the source code. Still, depending on how this feature would eventually be used in the library, the caller function must do it; also, the caller function needs to implement a mechanism to skip the associated region, according to the return value of __kmpc_start_record_task. Reviewed By: tianshilei1992 Differential Revision: https://reviews.llvm.org/D146642
51 lines
1.8 KiB
CMake
51 lines
1.8 KiB
CMake
# CMakeLists.txt file for unit testing OpenMP host runtime library.
|
|
include(CheckFunctionExists)
|
|
include(CheckLibraryExists)
|
|
|
|
# Some tests use math functions
|
|
check_library_exists(m sqrt "" LIBOMP_HAVE_LIBM)
|
|
# When using libgcc, -latomic may be needed for atomics
|
|
# (but when using compiler-rt, the atomics will be built-in)
|
|
# Note: we can not check for __atomic_load because clang treats it
|
|
# as special built-in and that breaks CMake checks
|
|
check_function_exists(__atomic_load_1 LIBOMP_HAVE_BUILTIN_ATOMIC)
|
|
if(NOT LIBOMP_HAVE_BUILTIN_ATOMIC)
|
|
check_library_exists(atomic __atomic_load_1 "" LIBOMP_HAVE_LIBATOMIC)
|
|
else()
|
|
# not needed
|
|
set(LIBOMP_HAVE_LIBATOMIC 0)
|
|
endif()
|
|
|
|
macro(pythonize_bool var)
|
|
if (${var})
|
|
set(${var} True)
|
|
else()
|
|
set(${var} False)
|
|
endif()
|
|
endmacro()
|
|
|
|
list(APPEND OPENMP_TEST_COMPILER_FEATURE_LIST "${LIBOMP_ARCH}")
|
|
update_test_compiler_features()
|
|
|
|
pythonize_bool(LIBOMP_USE_HWLOC)
|
|
pythonize_bool(LIBOMP_OMPT_SUPPORT)
|
|
pythonize_bool(LIBOMP_OMPT_OPTIONAL)
|
|
pythonize_bool(LIBOMP_OMPX_TASKGRAPH)
|
|
pythonize_bool(LIBOMP_HAVE_LIBM)
|
|
pythonize_bool(LIBOMP_HAVE_LIBATOMIC)
|
|
pythonize_bool(OPENMP_STANDALONE_BUILD)
|
|
pythonize_bool(OPENMP_TEST_COMPILER_HAS_OMIT_FRAME_POINTER_FLAGS)
|
|
pythonize_bool(OPENMP_TEST_COMPILER_HAS_OMP_H)
|
|
|
|
add_library(ompt-print-callback INTERFACE)
|
|
target_include_directories(ompt-print-callback INTERFACE ${CMAKE_CURRENT_SOURCE_DIR}/ompt)
|
|
|
|
|
|
add_openmp_testsuite(check-libomp "Running libomp tests" ${CMAKE_CURRENT_BINARY_DIR} DEPENDS omp)
|
|
# Add target check-ompt, but make sure to not add the tests twice to check-openmp.
|
|
add_openmp_testsuite(check-ompt "Running OMPT tests" ${CMAKE_CURRENT_BINARY_DIR}/ompt EXCLUDE_FROM_CHECK_ALL DEPENDS omp)
|
|
|
|
# Configure the lit.site.cfg.in file
|
|
set(AUTO_GEN_COMMENT "## Autogenerated by libomp configuration.\n# Do not edit!")
|
|
configure_file(lit.site.cfg.in lit.site.cfg @ONLY)
|