SDK: OMPT Support (#22)
* Ability to select alternative compiler per file
Implementation of ompt interface to rocprofiler SDK. task_create and task_schedule are not supported.
Misc updates
Update OpenMP target sample
- samples/ompt -> samples/openmp_target
- fix sample test of openmp-target
- reorganize files
Rework OpenMP implementation
Minor OpenMP implementation cleanup
Rename samples/openmp_target CMake targets
Add tests/bin/openmp
- OpenMP target test app in tests/bin/openmp/target
Format samples/openmp_target CMakeLists.txt
Misc lib/rocprofiler-sdk/openmp cleanup
- fix includes
- convert_arg
Update openmp.def.cpp
- tweak includes
- remove lots of temporary variables
Update samples
- common::get_callback_id_names() -> common::get_callback_tracing_names()
- add kernel dispatch, memory copy, scratch memory buffered tracing to openmp target sample
Fix code object operation names
- add "CODE_OBJECT_" prefix
Update include/rocprofiler-sdk/openmp/api_id.h
- remove spurious comment
Miscellaneous openmp updates
- similar API for openmp_begin and openmp_end
- move implementations of ompt callbacks to openmp.cpp
- ompt_{thread_begin,thread_end,parallel_begin,parallel_end}_callbacks are openmp_events
[SWDEV-484495] Fix int truncation in CSV output (#1098)
CSV output truncates doubles to ints when it shouldn't. Derived metrics
are (mostly) doubles and lose precision (or become worthless) if treated
as an int. Converted these to double to match the format we return from
rocprof-sdk.
Co-authored-by: Benjamin Welton <ben@amd.com>
Update limit for max counter records in rocprof-tool (#1073)
A fixed sized std::array is used to store counter records in rocprofiler SDK. This limit was breached in SWDEV-484742. Upping the limit to 512 to be less likely to reach this limit again.
adding proxy ompt_data_t * arguments
fixes for proxy pointers
- Implement proxy ompt_data_t* pointers for clients
- Add ompt_data_t* arguments back to callback API
- Modify openmp sample to illustrate use of proxy pointers
formatting
SWDEV-467350: Skipping tool counter iteration for unsupported hardware (#1083)
Fixing some accumulate metrics (#1089)
* Fixing some accumulate metrics
* Fixing some more accumulate metrics
---------
Co-authored-by: Benjamin Welton <bewelton@amd.com>
updating rocprofv3 help options (#1113)
* updating rocprofv3 help options
* updating CHANGELOG
Fixing installed pacakge tests in CI (#1119)
* Fixing installed pacakge tests in CI
* Formatted rocprofv3.py with black formatter
SWDEV-488948: PC Sampling - Correlation class to provide some thread safety. Adding multithread tests. (#1112)
* SWDEV-488948: PC Sampling - Correlation class to provide some thread safety. Adding multithread tests.
* Update source/lib/rocprofiler-sdk/pc_sampling/parser/correlation.hpp
Co-authored-by: Vladimir Indic <139573562+vlaindic@users.noreply.github.com>
* Update source/lib/rocprofiler-sdk/pc_sampling/parser/correlation.hpp
Co-authored-by: Vladimir Indic <139573562+vlaindic@users.noreply.github.com>
* Adding backlog for codeobj changes
* Formatting
* Update source/lib/rocprofiler-sdk/pc_sampling/code_object.hpp
Co-authored-by: Vladimir Indic <139573562+vlaindic@users.noreply.github.com>
* Update source/lib/rocprofiler-sdk/pc_sampling/code_object.hpp
Co-authored-by: Vladimir Indic <139573562+vlaindic@users.noreply.github.com>
---------
Co-authored-by: Vladimir Indic <139573562+vlaindic@users.noreply.github.com>
SWDEV-487621: Fixes for metric definitions (#1118)
* Fixes for metric definitions
* Removing gfx8
* Update changelog
* Fixing unit tests
* Small fixes
* Fix for write size
Fix PSDB change (#1120)
Reverts change to `source/include/rocprofiler-sdk/callback_tracing.h`
from commit 9b2ece76c3
clang-18 build fix for RCCL (#1123)
Removes ambiguity on const usage, which clang-18 complains about
(preventing build with warn error).
mem copy direction field update (#1124)
Adding Node-id for debugging with log level trace (#1090)
fix botched rebase
Per Jonathan to remove -rdynamic warning so CI will continue
pedantic formatting
Correct the package name of rocprofiler-sdk (#1126)
* Correct the package name of rocprofiler-sdk
ROCM VERSION(for ex: 60300) was missing in the package name.
Added the same
* Use cmake cache string while setting the variable for ROCm Version
* correct the cmake-format
---------
Co-authored-by: Ranjith Ramakrishnan <Ranjith.Ramakrishnan@amd.com>
Fixing kokkosp tool library packaging (#1121)
* Fixing kokkosp tool library packaging
* Update source/lib/rocprofiler-sdk-tool/kokkosp/CMakeLists.txt
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
* Update CMakeLists.txt
* Update CMakeLists.txt
* Component Requirement in CPack
* Adding package dependency
* Update CMakeLists.txt
* Update rocprofiler_config_packaging.cmake
* Fix rocprofiler-sdk-tool-kokkosp BUILD/INSTALL RPATH
- CMAKE_INSTALL_LIBDIR doesn't help
* Add BUILD/INSTALL RPATH to rocprofv3-trigger-list-metrics
- fixes packaging issues
* Update packaging
- core depends on rocprofiler-sdk-roctx
- add CPACK_DEBIAN_PACKAGE_SHLIBDEPS_PRIVATE_DIRS to resolve inter-package dependencies
* Fix package depends version format
* Improve tests/rocprofv3/summary/validate logging
* Update CI workflow
- prioritize roctx package in Install Packages step
* Remove setting <package-name>_VERSION in config.cmake.in
- this is automatically handled by existence of <package-name>-config-version.cmake
* Update rocprofiler-sdk-config.cmake
- relax find_package versioning requirements to same major and minor version
* Update rocprofiler-sdk-config.cmake
- relax find_package versioning requirements (remove EXACT, specify range)
* Tweak CI workflow
* Update perfetto_reader.py
- better handle failure to load trace processor
* Misc cleanup for config packaging
* Update config packaging
* Update config packaging
* Revert perfetto for core-rpm packages
* Revert perfetto for core-rpm packages
- perfetto < 0.9.0
* Tweak tests/rocprofv3/summary/validate.py
- reorder some checks
---------
Co-authored-by: Ammar Elwazir <aelwazir@useocpm2m-387-013.amd.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jonathan R. Madsen <jonathanrmadsen@gmail.com>
Clang Warning Fixes (#1131)
Builds prevented on clang-18
Adding start and end timestamp columns in csv (#1128)
* Adding start and end timestamp columns in csv
* Adding assert check for the counter timestamps
---------
Co-authored-by: Gopesh Bhardwaj <gopesh.bhardwaj@amd.com>
rocprofv3: docs and help menu updates (#1129)
* doc updates
* Correcting ROCtx information
* Making ROCTx string consistent
* missing occurence
Renamed agent profiling service to device counting service (#1132)
* Renamed agent profiling service to device counting service
Name more aptly represents what agent profiling did (device wide
counter collection). Conversion of existing user code can be
performed by the following find/sed command:
find . -type f -exec sed -i 's/rocprofiler_agent_profile_callback_t/rocprofiler_device_counting_service_callback_t/g; s/rocprofiler_configure_agent_profile_counting_service/rocprofiler_configure_device_counting_service/g; s/agent_profile.h/device_counting_service.h/g; s/rocprofiler_sample_agent_profile_counting_service/rocprofiler_sample_device_counting_service/g' {} +
* Converted dispatch profile to dispatch counting service
* Debug for functioal counters test
* Minor changes for CI
* Minor fix
* More fixes for CI
* Update evaluate_ast.cpp
---------
Co-authored-by: Benjamin Welton <ben@amd.com>
Testing updated RPM dockers (#1136)
* Testing updated RPM dockers
* Trying to fix PSDB for test package dependency
Agent Profiling Fixes for Broken/Improper API Usage (#1122)
Prevent's multiple setups of agent profiling on the same agent.
Fixes agent read context to only read agents that were setup.
Prevent copy of agent profiling internal data struct and reset
hsa_signal on move to prevent inadvertant delete.
Simplifying PR template (#1139)
Implementation of ompt interface to rocprofiler SDK. task_create and task_schedule are not supported.
Fixing installed pacakge tests in CI (#1119)
* Fixing installed pacakge tests in CI
* Formatted rocprofv3.py with black formatter
Fix PSDB change (#1120)
Reverts change to `source/include/rocprofiler-sdk/callback_tracing.h`
from commit 9b2ece76c3
delete unused files
added arguments to some OMPT buffter records
* Fix cmake issues
Remove rocprofiler_ompt_finalize_tool
- a public API function is not necessary: should just finalize rocprofiler-sdk
Fix duplicate ROCPROFILER_{BUFFER,CALLBACK}_TRACING_KIND_STRING
Add lib/rocprofiler-sdk/ompt.hpp
- declares rocprofiler::sdk::finalize_ompt
Remove change to tests/rocprofv3/summary/conftest.py
Add set_fini_status(1) back to registration.cpp
Deleted uneeded files
Incoporate OpenMP code and sample
Fix merge issues with amd-staging
Add push_correlation_id for OpenMP tasking; improve debugability
fixup bad merge
* Suppress OpenMP data race
* Fix openmp_target sample
* Enum and struct name changes + source code reorg
- remove mix of ompt and openmp
- opted for ompt
- changes made for consistency
- ompt_api -> ompt
- openmp_api -> ompt
- OPENMP -> OMPT
* Update tests and more renaming
- dest_device_num -> dst_device_num
- src_addr -> src_address
- dest_addr -> dst_address
- remove info_type::begin
- require OMP_TARGET_OFFLOAD
* Update openmp-target test/sample env and labels
* Formatting
* Tweaks to cmake for openmp target
- Disable for thread sanitizers due to preloading issue
* OpenMP target cmake updates
- remove gfx1010 (fails on mi300)
- OPENMP_GPU_TARGETS
* Remove device_unload and target_map_emi support
- these are never supported by AMD OpenMP compilers
* Update CI workflow
- exclude openmp-target tests from navi3 and vega20
---------
Co-authored-by: Larry Meadows <Lawrence.Meadows@amd.com>
Co-authored-by: Jonathan R. Madsen <jonathanrmadsen@gmail.com>
Этот коммит содержится в:
Madsen, Jonathan
коммит произвёл
GitHub
GitHub
родитель
a579c70b71
Коммит
00c46fd5e5
@@ -27,7 +27,6 @@ set(ROCPROFILER_HEADER_FILES
|
||||
internal_threading.h
|
||||
marker.h
|
||||
ompt.h
|
||||
openmp.h
|
||||
pc_sampling.h
|
||||
profile_config.h
|
||||
registration.h
|
||||
@@ -43,7 +42,7 @@ install(
|
||||
add_subdirectory(hip)
|
||||
add_subdirectory(hsa)
|
||||
add_subdirectory(marker)
|
||||
add_subdirectory(openmp)
|
||||
add_subdirectory(ompt)
|
||||
add_subdirectory(rccl)
|
||||
add_subdirectory(cxx)
|
||||
add_subdirectory(kfd)
|
||||
|
||||
@@ -84,7 +84,7 @@ typedef struct
|
||||
} rocprofiler_buffer_tracing_hip_api_record_t;
|
||||
|
||||
/**
|
||||
* @brief Additional trace data for OpenMP target routines
|
||||
* @brief Additional trace data for OMPT target routines
|
||||
*/
|
||||
|
||||
typedef struct rocprofiler_buffer_tracing_ompt_target_t
|
||||
@@ -98,13 +98,13 @@ typedef struct rocprofiler_buffer_tracing_ompt_target_t
|
||||
|
||||
typedef struct rocprofiler_buffer_tracing_ompt_target_data_op_t
|
||||
{
|
||||
uint64_t host_op_id; // from the host_op_id argument to the OMPT callback
|
||||
int32_t optype; // ompt_target_data_op_t kind of operation
|
||||
int32_t src_device_num; // ompt device number for data source
|
||||
int32_t dest_device_num; // ompt device number for data destination
|
||||
int32_t reserved; // for padding
|
||||
uint64_t bytes; // size in bytes of the operation
|
||||
const void* codeptr_ra; // pointer to the callsite of the target_data_op
|
||||
uint64_t host_op_id; // from the host_op_id argument to the OMPT callback
|
||||
int32_t optype; // ompt_target_data_op_t kind of operation
|
||||
int32_t src_device_num; // ompt device number for data source
|
||||
int32_t dst_device_num; // ompt device number for data destination
|
||||
int32_t reserved; // for padding
|
||||
uint64_t bytes; // size in bytes of the operation
|
||||
const void* codeptr_ra; // pointer to the callsite of the target_data_op
|
||||
} rocprofiler_buffer_tracing_ompt_target_data_op_t;
|
||||
|
||||
typedef struct rocprofiler_buffer_tracing_ompt_target_kernel_t
|
||||
@@ -115,9 +115,9 @@ typedef struct rocprofiler_buffer_tracing_ompt_target_kernel_t
|
||||
} rocprofiler_buffer_tracing_ompt_target_kernel_t;
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler Buffer OPENMP API Tracer Record.
|
||||
* @brief ROCProfiler Buffer OMPT Tracer Record.
|
||||
*/
|
||||
typedef struct rocprofiler_buffer_tracing_ompt_api_record_t
|
||||
typedef struct rocprofiler_buffer_tracing_ompt_record_t
|
||||
{
|
||||
uint64_t size; ///< size of this struct
|
||||
rocprofiler_buffer_tracing_kind_t kind;
|
||||
@@ -129,16 +129,16 @@ typedef struct rocprofiler_buffer_tracing_ompt_api_record_t
|
||||
union
|
||||
{
|
||||
rocprofiler_buffer_tracing_ompt_target_t target;
|
||||
rocprofiler_buffer_tracing_ompt_target_data_op_t target_data;
|
||||
rocprofiler_buffer_tracing_ompt_target_kernel_t kernel;
|
||||
rocprofiler_buffer_tracing_ompt_target_data_op_t target_data_op;
|
||||
rocprofiler_buffer_tracing_ompt_target_kernel_t target_kernel;
|
||||
uint64_t reserved[5];
|
||||
};
|
||||
|
||||
/// @var kind
|
||||
/// @brief ::ROCPROFILER_CALLBACK_TRACING_OPENMP
|
||||
/// @brief ::ROCPROFILER_BUFFER_TRACING_OMPT
|
||||
/// @var operation
|
||||
/// @brief Specification of the API function,::rocprofiler_ompt_operation_t
|
||||
} rocprofiler_buffer_tracing_ompt_api_record_t;
|
||||
/// @brief Specification of the ::rocprofiler_ompt_operation_t
|
||||
} rocprofiler_buffer_tracing_ompt_record_t;
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler Buffer Marker Tracer Record.
|
||||
|
||||
@@ -27,7 +27,7 @@
|
||||
#include <rocprofiler-sdk/hip.h>
|
||||
#include <rocprofiler-sdk/hsa.h>
|
||||
#include <rocprofiler-sdk/marker.h>
|
||||
#include <rocprofiler-sdk/openmp.h>
|
||||
#include <rocprofiler-sdk/ompt.h>
|
||||
#include <rocprofiler-sdk/rccl.h>
|
||||
|
||||
#include <hsa/hsa.h>
|
||||
@@ -80,13 +80,13 @@ typedef struct
|
||||
} rocprofiler_callback_tracing_hip_api_data_t;
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler OPENMP Callback Data
|
||||
* @brief ROCProfiler OMPT Callback Data
|
||||
*/
|
||||
typedef struct
|
||||
{
|
||||
uint64_t size; ///< size of this struct
|
||||
rocprofiler_ompt_api_args_t args;
|
||||
} rocprofiler_callback_tracing_ompt_api_data_t;
|
||||
uint64_t size; ///< size of this struct
|
||||
rocprofiler_ompt_args_t args;
|
||||
} rocprofiler_callback_tracing_ompt_data_t;
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler Marker Tracer Callback Data.
|
||||
|
||||
@@ -78,6 +78,7 @@ ROCPROFILER_DEFINE_CATEGORY(category, hsa_api, "HSA API function")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, hip_api, "HIP API function")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, marker_api, "Marker API region")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, rccl_api, "RCCL API function")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, openmp, "OpenMP")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, kernel_dispatch, "GPU kernel dispatch")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, memory_copy, "Async memory copy")
|
||||
ROCPROFILER_DEFINE_CATEGORY(category, memory_allocation, "Memory Allocation")
|
||||
@@ -87,6 +88,7 @@ ROCPROFILER_DEFINE_CATEGORY(category, memory_allocation, "Memory Allocation")
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::hip_api), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::marker_api), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::rccl_api), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::openmp), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::kernel_dispatch), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::memory_copy), \
|
||||
ROCPROFILER_PERFETTO_CATEGORY(category::memory_allocation)
|
||||
|
||||
@@ -358,6 +358,14 @@ save(ArchiveT& ar, rocprofiler_callback_tracing_rccl_api_data_t data)
|
||||
ROCP_SDK_SAVE_DATA_FIELD(retval);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_callback_tracing_ompt_data_t data)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_FIELD(size);
|
||||
// ROCP_SDK_SAVE_DATA_FIELD(args);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_dispatch_counting_service_data_t data)
|
||||
@@ -443,6 +451,56 @@ save(ArchiveT& ar, rocprofiler_buffer_tracing_rccl_api_record_t data)
|
||||
save_buffer_tracing_api_record(ar, data);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_buffer_tracing_ompt_target_t data)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_VALUE("kind", kind);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("device", device_num);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("task_id", task_id);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("target_id", target_id);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_buffer_tracing_ompt_target_data_op_t data)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_VALUE("host_op_id", host_op_id);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("optype", optype);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("src_device_num", src_device_num);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("dst_device_num", dst_device_num);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("bytes", bytes);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_buffer_tracing_ompt_target_kernel_t data)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_VALUE("host_op_id", host_op_id);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("device_num", device_num);
|
||||
ROCP_SDK_SAVE_DATA_VALUE("requested_num_teams", requested_num_teams);
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_buffer_tracing_ompt_record_t data)
|
||||
{
|
||||
save_buffer_tracing_api_record(ar, data);
|
||||
|
||||
if(data.operation == ROCPROFILER_OMPT_ID_target_emi)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_FIELD(target);
|
||||
}
|
||||
else if(data.operation == ROCPROFILER_OMPT_ID_target_data_op_emi)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_FIELD(target_data_op);
|
||||
}
|
||||
else if(data.operation == ROCPROFILER_OMPT_ID_target_submit_emi)
|
||||
{
|
||||
ROCP_SDK_SAVE_DATA_FIELD(target_kernel);
|
||||
}
|
||||
}
|
||||
|
||||
template <typename ArchiveT>
|
||||
void
|
||||
save(ArchiveT& ar, rocprofiler_buffer_tracing_kernel_dispatch_record_t data)
|
||||
|
||||
@@ -67,7 +67,7 @@ typedef enum // NOLINT(performance-enum-size)
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_KERNEL_DISPATCH, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_SCRATCH_MEMORY, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_RCCL_API, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_OPENMP, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_OMPT, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_MEMORY_ALLOCATION, ///<
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_LAST,
|
||||
} rocprofiler_external_correlation_id_request_kind_t;
|
||||
|
||||
@@ -170,7 +170,7 @@ typedef enum // NOLINT(performance-enum-size)
|
||||
ROCPROFILER_CALLBACK_TRACING_KERNEL_DISPATCH, ///< Callbacks for kernel dispatches
|
||||
ROCPROFILER_CALLBACK_TRACING_MEMORY_COPY, ///< @see ::rocprofiler_memory_copy_operation_t
|
||||
ROCPROFILER_CALLBACK_TRACING_RCCL_API, ///< @RCCL tracing
|
||||
ROCPROFILER_CALLBACK_TRACING_OPENMP, ///< @see ::rocprofiler_ompt_operation_t
|
||||
ROCPROFILER_CALLBACK_TRACING_OMPT, ///< @see ::rocprofiler_ompt_operation_t
|
||||
ROCPROFILER_CALLBACK_TRACING_MEMORY_ALLOCATION, ///< @see
|
||||
///< ::rocprofiler_memory_allocation_operation_t
|
||||
ROCPROFILER_CALLBACK_TRACING_RUNTIME_INITIALIZATION, ///< Callback notifying that a runtime
|
||||
@@ -200,7 +200,7 @@ typedef enum // NOLINT(performance-enum-size)
|
||||
ROCPROFILER_BUFFER_TRACING_SCRATCH_MEMORY, ///< Buffer scratch memory reclaimation info
|
||||
ROCPROFILER_BUFFER_TRACING_CORRELATION_ID_RETIREMENT, ///< Correlation ID in no longer in use
|
||||
ROCPROFILER_BUFFER_TRACING_RCCL_API, ///< RCCL tracing
|
||||
ROCPROFILER_BUFFER_TRACING_OPENMP, ///< @see ::rocprofiler_ompt_operation_t
|
||||
ROCPROFILER_BUFFER_TRACING_OMPT, ///< @see ::rocprofiler_ompt_operation_t
|
||||
ROCPROFILER_BUFFER_TRACING_MEMORY_ALLOCATION, ///< @see
|
||||
///< ::rocprofiler_memory_allocation_operation_t
|
||||
ROCPROFILER_BUFFER_TRACING_RUNTIME_INITIALIZATION, ///< Record indicating a runtime library has
|
||||
|
||||
@@ -24,13 +24,14 @@
|
||||
|
||||
#include <rocprofiler-sdk/defines.h>
|
||||
#include <rocprofiler-sdk/fwd.h>
|
||||
#include <rocprofiler-sdk/openmp/omp-tools.h>
|
||||
#include <rocprofiler-sdk/registration.h>
|
||||
#include <rocprofiler-sdk/ompt/api_args.h>
|
||||
#include <rocprofiler-sdk/ompt/api_id.h>
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
|
||||
/**
|
||||
* @defgroup OMPT_REGISTRATION Tool registration for OpenMP Tools
|
||||
*
|
||||
* Functions for enabling OpenMP support in tools which provide their own ompt_start_tool symbol but
|
||||
* Functions for enabling OMPT support in tools which provide their own ompt_start_tool symbol but
|
||||
* want to defer to rocprofiler-sdk for OMPT.
|
||||
*
|
||||
* @{
|
||||
@@ -38,9 +39,6 @@
|
||||
|
||||
ROCPROFILER_EXTERN_C_INIT
|
||||
|
||||
void
|
||||
rocprofiler_ompt_finalize_tool() ROCPROFILER_API;
|
||||
|
||||
rocprofiler_status_t
|
||||
rocprofiler_ompt_is_initialized(int* status) ROCPROFILER_API ROCPROFILER_NONNULL(1);
|
||||
|
||||
|
||||
@@ -0,0 +1,11 @@
|
||||
#
|
||||
#
|
||||
# Installation of public OMPT headers
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_OMPT_HEADER_FILES api_args.h api_id.h omp-tools.h)
|
||||
|
||||
install(
|
||||
FILES ${ROCPROFILER_OMPT_HEADER_FILES}
|
||||
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/rocprofiler-sdk/ompt
|
||||
COMPONENT development)
|
||||
+20
-20
@@ -25,7 +25,7 @@
|
||||
#include <rocprofiler-sdk/defines.h>
|
||||
#include <rocprofiler-sdk/version.h>
|
||||
|
||||
#include <rocprofiler-sdk/openmp/omp-tools.h>
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
@@ -60,7 +60,7 @@ typedef struct rocprofiler_ompt_no_args
|
||||
char empty;
|
||||
} rocprofiler_ompt_no_args;
|
||||
|
||||
typedef union rocprofiler_ompt_api_args_t
|
||||
typedef union rocprofiler_ompt_args_t
|
||||
{
|
||||
// The ompt_data_t* values passed to the client tool are proxies.
|
||||
// This allows the client tool to use them as it would in their own
|
||||
@@ -150,11 +150,11 @@ typedef union rocprofiler_ompt_api_args_t
|
||||
uint64_t module_id;
|
||||
} device_load;
|
||||
|
||||
struct
|
||||
{
|
||||
int device_num;
|
||||
uint64_t module_id;
|
||||
} device_unload;
|
||||
// struct
|
||||
// {
|
||||
// int device_num;
|
||||
// uint64_t module_id;
|
||||
// } device_unload;
|
||||
|
||||
struct
|
||||
{
|
||||
@@ -299,10 +299,10 @@ typedef union rocprofiler_ompt_api_args_t
|
||||
ompt_data_t* target_data;
|
||||
ompt_data_t* host_op_id;
|
||||
ompt_target_data_op_t optype;
|
||||
void* src_addr;
|
||||
void* src_address;
|
||||
int src_device_num;
|
||||
void* dest_addr;
|
||||
int dest_device_num;
|
||||
void* dst_address;
|
||||
int dst_device_num;
|
||||
size_t bytes;
|
||||
const void* codeptr_ra;
|
||||
} target_data_op_emi;
|
||||
@@ -315,15 +315,15 @@ typedef union rocprofiler_ompt_api_args_t
|
||||
unsigned int requested_num_teams;
|
||||
} target_submit_emi;
|
||||
|
||||
struct
|
||||
{
|
||||
unsigned int nitems;
|
||||
void** host_addr;
|
||||
void** device_addr;
|
||||
size_t* bytes;
|
||||
unsigned int* mapping_flags;
|
||||
const void* codeptr_ra;
|
||||
} target_map_emi;
|
||||
// struct
|
||||
// {
|
||||
// unsigned int nitems;
|
||||
// void** host_addr;
|
||||
// void** device_addr;
|
||||
// size_t* bytes;
|
||||
// unsigned int* mapping_flags;
|
||||
// const void* codeptr_ra;
|
||||
// } target_map_emi;
|
||||
|
||||
struct
|
||||
{
|
||||
@@ -335,6 +335,6 @@ typedef union rocprofiler_ompt_api_args_t
|
||||
|
||||
rocprofiler_ompt_callback_functions_t callback_functions;
|
||||
|
||||
} rocprofiler_ompt_api_args_t;
|
||||
} rocprofiler_ompt_args_t;
|
||||
|
||||
ROCPROFILER_EXTERN_C_FINI
|
||||
@@ -0,0 +1,67 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler enumeration of OMPT (OpenMP tools) tracing operations
|
||||
* NOTE: These are callbacks into the ROCProfiler SDK from the vendor-provided OMPT implementation
|
||||
*/
|
||||
typedef enum // NOLINT(performance-enum-size)
|
||||
{
|
||||
ROCPROFILER_OMPT_ID_NONE = -1,
|
||||
ROCPROFILER_OMPT_ID_thread_begin = 0,
|
||||
ROCPROFILER_OMPT_ID_thread_end,
|
||||
ROCPROFILER_OMPT_ID_parallel_begin,
|
||||
ROCPROFILER_OMPT_ID_parallel_end,
|
||||
ROCPROFILER_OMPT_ID_task_create,
|
||||
ROCPROFILER_OMPT_ID_task_schedule,
|
||||
ROCPROFILER_OMPT_ID_implicit_task,
|
||||
ROCPROFILER_OMPT_ID_device_initialize,
|
||||
ROCPROFILER_OMPT_ID_device_finalize,
|
||||
ROCPROFILER_OMPT_ID_device_load,
|
||||
// ROCPROFILER_OMPT_ID_device_unload,
|
||||
ROCPROFILER_OMPT_ID_sync_region_wait,
|
||||
ROCPROFILER_OMPT_ID_mutex_released,
|
||||
ROCPROFILER_OMPT_ID_dependences,
|
||||
ROCPROFILER_OMPT_ID_task_dependence,
|
||||
ROCPROFILER_OMPT_ID_work,
|
||||
ROCPROFILER_OMPT_ID_masked,
|
||||
ROCPROFILER_OMPT_ID_sync_region,
|
||||
ROCPROFILER_OMPT_ID_lock_init,
|
||||
ROCPROFILER_OMPT_ID_lock_destroy,
|
||||
ROCPROFILER_OMPT_ID_mutex_acquire,
|
||||
ROCPROFILER_OMPT_ID_mutex_acquired,
|
||||
ROCPROFILER_OMPT_ID_nest_lock,
|
||||
ROCPROFILER_OMPT_ID_flush,
|
||||
ROCPROFILER_OMPT_ID_cancel,
|
||||
ROCPROFILER_OMPT_ID_reduction,
|
||||
ROCPROFILER_OMPT_ID_dispatch,
|
||||
ROCPROFILER_OMPT_ID_target_emi,
|
||||
ROCPROFILER_OMPT_ID_target_data_op_emi,
|
||||
ROCPROFILER_OMPT_ID_target_submit_emi,
|
||||
// ROCPROFILER_OMPT_ID_target_map_emi,
|
||||
ROCPROFILER_OMPT_ID_error,
|
||||
ROCPROFILER_OMPT_ID_callback_functions, // fake to return struct of ompt callback function
|
||||
// pointers
|
||||
ROCPROFILER_OMPT_ID_LAST
|
||||
} rocprofiler_ompt_operation_t;
|
||||
@@ -1,11 +0,0 @@
|
||||
#
|
||||
#
|
||||
# Installation of public OpenMP headers
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_OPENMP_HEADER_FILES api_args.h api_id.h omp-tools.h)
|
||||
|
||||
install(
|
||||
FILES ${ROCPROFILER_OPENMP_HEADER_FILES}
|
||||
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/rocprofiler-sdk/openmp
|
||||
COMPONENT development)
|
||||
@@ -1,67 +0,0 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
/**
|
||||
* @brief ROCProfiler enumeration of OPENMP (OpenMP tools) tracing operations
|
||||
* NOTE: These are callbacks into the ROCProfiler SDK from the vendor-provided OMPT implementation
|
||||
*/
|
||||
typedef enum // NOLINT(performance-enum-size)
|
||||
{
|
||||
ROCPROFILER_OPENMP_ID_NONE = -1,
|
||||
ROCPROFILER_OPENMP_ID_thread_begin = 0,
|
||||
ROCPROFILER_OPENMP_ID_thread_end,
|
||||
ROCPROFILER_OPENMP_ID_parallel_begin,
|
||||
ROCPROFILER_OPENMP_ID_parallel_end,
|
||||
ROCPROFILER_OPENMP_ID_task_create,
|
||||
ROCPROFILER_OPENMP_ID_task_schedule,
|
||||
ROCPROFILER_OPENMP_ID_implicit_task,
|
||||
ROCPROFILER_OPENMP_ID_device_initialize,
|
||||
ROCPROFILER_OPENMP_ID_device_finalize,
|
||||
ROCPROFILER_OPENMP_ID_device_load,
|
||||
ROCPROFILER_OPENMP_ID_device_unload,
|
||||
ROCPROFILER_OPENMP_ID_sync_region_wait,
|
||||
ROCPROFILER_OPENMP_ID_mutex_released,
|
||||
ROCPROFILER_OPENMP_ID_dependences,
|
||||
ROCPROFILER_OPENMP_ID_task_dependence,
|
||||
ROCPROFILER_OPENMP_ID_work,
|
||||
ROCPROFILER_OPENMP_ID_masked,
|
||||
ROCPROFILER_OPENMP_ID_sync_region,
|
||||
ROCPROFILER_OPENMP_ID_lock_init,
|
||||
ROCPROFILER_OPENMP_ID_lock_destroy,
|
||||
ROCPROFILER_OPENMP_ID_mutex_acquire,
|
||||
ROCPROFILER_OPENMP_ID_mutex_acquired,
|
||||
ROCPROFILER_OPENMP_ID_nest_lock,
|
||||
ROCPROFILER_OPENMP_ID_flush,
|
||||
ROCPROFILER_OPENMP_ID_cancel,
|
||||
ROCPROFILER_OPENMP_ID_reduction,
|
||||
ROCPROFILER_OPENMP_ID_dispatch,
|
||||
ROCPROFILER_OPENMP_ID_target_emi,
|
||||
ROCPROFILER_OPENMP_ID_target_data_op_emi,
|
||||
ROCPROFILER_OPENMP_ID_target_submit_emi,
|
||||
ROCPROFILER_OPENMP_ID_target_map_emi,
|
||||
ROCPROFILER_OPENMP_ID_error,
|
||||
ROCPROFILER_OPENMP_ID_callback_functions, // fake to return struct of ompt callback function
|
||||
// pointers
|
||||
ROCPROFILER_OPENMP_ID_LAST
|
||||
} rocprofiler_ompt_operation_t;
|
||||
@@ -5,7 +5,7 @@ rocprofiler_activate_clang_tidy()
|
||||
|
||||
set(ROCPROFILER_LIB_HEADERS
|
||||
agent.hpp buffer.hpp external_correlation.hpp intercept_table.hpp
|
||||
internal_threading.hpp registration.hpp runtime_initialization.hpp)
|
||||
internal_threading.hpp ompt.hpp registration.hpp runtime_initialization.hpp)
|
||||
set(ROCPROFILER_LIB_SOURCES
|
||||
agent.cpp
|
||||
buffer.cpp
|
||||
@@ -18,10 +18,11 @@ set(ROCPROFILER_LIB_SOURCES
|
||||
external_correlation.cpp
|
||||
intercept_table.cpp
|
||||
internal_threading.cpp
|
||||
ompt.cpp
|
||||
pc_sampling.cpp
|
||||
profile_config.cpp
|
||||
registration.cpp
|
||||
rocprofiler.cpp
|
||||
registration.cpp
|
||||
runtime_initialization.cpp)
|
||||
|
||||
# ----------------------------------------------------------------------------------------#
|
||||
@@ -52,6 +53,7 @@ add_subdirectory(kernel_dispatch)
|
||||
add_subdirectory(page_migration)
|
||||
add_subdirectory(rccl)
|
||||
add_subdirectory(details)
|
||||
add_subdirectory(ompt)
|
||||
|
||||
target_link_libraries(
|
||||
rocprofiler-sdk-object-library
|
||||
|
||||
@@ -30,6 +30,7 @@
|
||||
#include "lib/rocprofiler-sdk/hsa/scratch_memory.hpp"
|
||||
#include "lib/rocprofiler-sdk/kernel_dispatch/kernel_dispatch.hpp"
|
||||
#include "lib/rocprofiler-sdk/marker/marker.hpp"
|
||||
#include "lib/rocprofiler-sdk/ompt/ompt.hpp"
|
||||
#include "lib/rocprofiler-sdk/page_migration/page_migration.hpp"
|
||||
#include "lib/rocprofiler-sdk/rccl/rccl.hpp"
|
||||
#include "lib/rocprofiler-sdk/registration.hpp"
|
||||
@@ -88,7 +89,7 @@ ROCPROFILER_BUFFER_TRACING_KIND_STRING(PAGE_MIGRATION)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(SCRATCH_MEMORY)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(CORRELATION_ID_RETIREMENT)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(RCCL_API)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(OPENMP)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(OMPT)
|
||||
ROCPROFILER_BUFFER_TRACING_KIND_STRING(RUNTIME_INITIALIZATION)
|
||||
|
||||
template <size_t Idx, size_t... Tail>
|
||||
@@ -273,6 +274,11 @@ rocprofiler_query_buffer_tracing_kind_operation_name(rocprofiler_buffer_tracing_
|
||||
val = rocprofiler::page_migration::name_by_id(operation);
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_OMPT:
|
||||
{
|
||||
val = rocprofiler::ompt::name_by_id(operation);
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_RUNTIME_INITIALIZATION:
|
||||
{
|
||||
val = rocprofiler::runtime_init::name_by_id(operation);
|
||||
@@ -282,10 +288,6 @@ rocprofiler_query_buffer_tracing_kind_operation_name(rocprofiler_buffer_tracing_
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_OPENMP:
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
}
|
||||
};
|
||||
|
||||
if(!val)
|
||||
@@ -403,6 +405,11 @@ rocprofiler_iterate_buffer_tracing_kind_operations(
|
||||
ops = rocprofiler::page_migration::get_ids();
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_OMPT:
|
||||
{
|
||||
ops = rocprofiler::ompt::get_ids();
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_RUNTIME_INITIALIZATION:
|
||||
{
|
||||
ops = rocprofiler::runtime_init::get_ids();
|
||||
@@ -412,10 +419,6 @@ rocprofiler_iterate_buffer_tracing_kind_operations(
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
}
|
||||
case ROCPROFILER_BUFFER_TRACING_OPENMP:
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
}
|
||||
}
|
||||
|
||||
for(const auto& itr : ops)
|
||||
|
||||
@@ -30,6 +30,7 @@
|
||||
#include "lib/rocprofiler-sdk/hsa/scratch_memory.hpp"
|
||||
#include "lib/rocprofiler-sdk/kernel_dispatch/kernel_dispatch.hpp"
|
||||
#include "lib/rocprofiler-sdk/marker/marker.hpp"
|
||||
#include "lib/rocprofiler-sdk/ompt/ompt.hpp"
|
||||
#include "lib/rocprofiler-sdk/rccl/rccl.hpp"
|
||||
#include "lib/rocprofiler-sdk/registration.hpp"
|
||||
#include "lib/rocprofiler-sdk/runtime_initialization.hpp"
|
||||
@@ -85,7 +86,7 @@ ROCPROFILER_CALLBACK_TRACING_KIND_STRING(KERNEL_DISPATCH)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(MEMORY_COPY)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(MEMORY_ALLOCATION)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(RCCL_API)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(OPENMP)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(OMPT)
|
||||
ROCPROFILER_CALLBACK_TRACING_KIND_STRING(RUNTIME_INITIALIZATION)
|
||||
|
||||
template <size_t Idx, size_t... Tail>
|
||||
@@ -255,9 +256,9 @@ rocprofiler_query_callback_tracing_kind_operation_name(rocprofiler_callback_trac
|
||||
val = rocprofiler::hsa::async_copy::name_by_id(operation);
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_OPENMP:
|
||||
case ROCPROFILER_CALLBACK_TRACING_OMPT:
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
val = rocprofiler::ompt::name_by_id(operation);
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_MEMORY_ALLOCATION:
|
||||
@@ -383,9 +384,9 @@ rocprofiler_iterate_callback_tracing_kind_operations(
|
||||
ops = rocprofiler::hsa::async_copy::get_ids();
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_OPENMP:
|
||||
case ROCPROFILER_CALLBACK_TRACING_OMPT:
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
ops = rocprofiler::ompt::get_ids();
|
||||
break;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_MEMORY_ALLOCATION:
|
||||
@@ -522,13 +523,22 @@ rocprofiler_iterate_callback_tracing_kind_operation_args(
|
||||
user_data);
|
||||
return ROCPROFILER_STATUS_SUCCESS;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_OMPT:
|
||||
{
|
||||
rocprofiler::ompt::iterate_args(
|
||||
record.operation,
|
||||
*static_cast<rocprofiler_callback_tracing_ompt_data_t*>(record.payload),
|
||||
callback,
|
||||
max_deref,
|
||||
user_data);
|
||||
return ROCPROFILER_STATUS_SUCCESS;
|
||||
}
|
||||
case ROCPROFILER_CALLBACK_TRACING_SCRATCH_MEMORY:
|
||||
case ROCPROFILER_CALLBACK_TRACING_CODE_OBJECT:
|
||||
case ROCPROFILER_CALLBACK_TRACING_KERNEL_DISPATCH:
|
||||
case ROCPROFILER_CALLBACK_TRACING_MEMORY_COPY:
|
||||
case ROCPROFILER_CALLBACK_TRACING_MEMORY_ALLOCATION:
|
||||
case ROCPROFILER_CALLBACK_TRACING_RCCL_API:
|
||||
case ROCPROFILER_CALLBACK_TRACING_OPENMP:
|
||||
case ROCPROFILER_CALLBACK_TRACING_RUNTIME_INITIALIZATION:
|
||||
{
|
||||
return ROCPROFILER_STATUS_ERROR_NOT_IMPLEMENTED;
|
||||
|
||||
@@ -66,7 +66,7 @@ struct code_object_info;
|
||||
struct code_object_info<ROCPROFILER_CODE_OBJECT_##OPERATION> \
|
||||
{ \
|
||||
static constexpr auto operation_idx = ROCPROFILER_CODE_OBJECT_##OPERATION; \
|
||||
static constexpr auto name = #OPERATION; \
|
||||
static constexpr auto name = "CODE_OBJECT_" #OPERATION; \
|
||||
};
|
||||
|
||||
SPECIALIZE_CODE_OBJECT_INFO(NONE)
|
||||
|
||||
@@ -152,20 +152,53 @@ pop_latest_correlation_id(correlation_id* val)
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
if(get_latest_correlation_id_impl().empty())
|
||||
auto& stack = get_latest_correlation_id_impl();
|
||||
if(stack.empty())
|
||||
{
|
||||
ROCP_ERROR << "empty thread-local correlation id stack";
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
ROCP_ERROR_IF(get_latest_correlation_id_impl().back() != val)
|
||||
<< "pop_latest_correlation_id is happening out of order for " << val->internal
|
||||
<< ". top of stack is " << get_latest_correlation_id_impl().back()->internal;
|
||||
if(stack.back() != val)
|
||||
{
|
||||
ROCP_ERROR << "pop_latest_correlation_id is happening out of order for " << val->internal
|
||||
<< ". top of stack is " << stack.back()->internal;
|
||||
}
|
||||
|
||||
get_latest_correlation_id_impl().pop_back();
|
||||
stack.pop_back();
|
||||
|
||||
return (get_latest_correlation_id_impl().empty()) ? nullptr
|
||||
: get_latest_correlation_id_impl().back();
|
||||
return (stack.empty()) ? nullptr : stack.back();
|
||||
}
|
||||
|
||||
correlation_id*
|
||||
push_correlation_id(correlation_id* val)
|
||||
{
|
||||
if(!val)
|
||||
{
|
||||
ROCP_ERROR << "passed nullptr to correlation id";
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
val->thread_idx = common::get_tid();
|
||||
get_latest_correlation_id_impl().emplace_back(val);
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
void
|
||||
dump_correlation_stack(const char* s)
|
||||
{
|
||||
auto& stack = get_latest_correlation_id_impl();
|
||||
auto info = std::stringstream{};
|
||||
info << s << ": tid: " << common::get_tid() << " :";
|
||||
for(const auto* itr : stack)
|
||||
{
|
||||
info << " " << itr->internal;
|
||||
;
|
||||
}
|
||||
info << "\n";
|
||||
printf("%s", info.str().c_str());
|
||||
}
|
||||
|
||||
} // namespace context
|
||||
} // namespace rocprofiler
|
||||
|
||||
@@ -86,6 +86,14 @@ get_latest_correlation_id();
|
||||
const correlation_id*
|
||||
pop_latest_correlation_id(correlation_id*);
|
||||
|
||||
// push correlation id
|
||||
correlation_id*
|
||||
push_correlation_id(correlation_id*);
|
||||
|
||||
// dump the cid stack for debugging
|
||||
void
|
||||
dump_correlation_stack(const char*);
|
||||
|
||||
/// permits tools opportunity to modify the correlation id based on the domain, op, and
|
||||
/// the rocprofiler generated correlation id
|
||||
struct correlation_tracing_service
|
||||
|
||||
@@ -0,0 +1,251 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in all
|
||||
// copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
// SOFTWARE.
|
||||
|
||||
#include "lib/rocprofiler-sdk/ompt/ompt.hpp"
|
||||
#include "lib/common/logging.hpp"
|
||||
#include "lib/common/static_object.hpp"
|
||||
#include "lib/rocprofiler-sdk/ompt/details/format.hpp" // NOLINT(unused-includes)
|
||||
#include "lib/rocprofiler-sdk/registration.hpp"
|
||||
|
||||
#include <rocprofiler-sdk/fwd.h>
|
||||
#include <rocprofiler-sdk/ompt.h>
|
||||
#include <rocprofiler-sdk/ompt/api_args.h>
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <cstdint>
|
||||
#include <iostream>
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace ompt
|
||||
{
|
||||
namespace
|
||||
{
|
||||
ompt_start_tool_result_t*
|
||||
get_start_tool_result()
|
||||
{
|
||||
static auto*& obj = common::static_object<ompt_start_tool_result_t>::construct();
|
||||
return obj;
|
||||
}
|
||||
|
||||
namespace
|
||||
{
|
||||
auto init_status = std::atomic<int>{0};
|
||||
auto fini_status = std::atomic<int>{0};
|
||||
ompt_finalize_tool_t tool_finalize = nullptr;
|
||||
ompt_set_callback_t set_callback = nullptr;
|
||||
rocprofiler_ompt_callback_functions_t ompt_callback_table = {};
|
||||
ompt_get_thread_data_t real_get_thread_data = nullptr;
|
||||
ompt_get_parallel_info_t real_get_parallel_info = nullptr;
|
||||
ompt_get_task_info_t real_get_task_info = nullptr;
|
||||
ompt_get_target_info_t real_get_target_info = nullptr;
|
||||
} // namespace
|
||||
|
||||
void
|
||||
set_ompt_callbacks()
|
||||
{
|
||||
// set all the ompt callbacks that might be used
|
||||
|
||||
auto cb = [](const char* cbname, ompt_callback_t* cbf, int cbnum) {
|
||||
ompt_set_result_t result = set_callback(static_cast<ompt_callbacks_t>(cbnum), *cbf);
|
||||
ROCP_WARNING_IF(result != ompt_set_always)
|
||||
<< "rocprofiler-sdk OpenMP tools set_callback returned " << result
|
||||
<< fmt::format(" (set result = {})", result) << " for " << cbname << " (id=" << cbnum
|
||||
<< ")";
|
||||
};
|
||||
ompt::update_table(cb);
|
||||
ompt::update_callback(ompt_callback_table);
|
||||
}
|
||||
|
||||
// proxies for some entry points that use ompt_data_t *
|
||||
ompt_data_t*
|
||||
proxy_get_thread_data()
|
||||
{
|
||||
ompt_data_t* real = real_get_thread_data();
|
||||
auto* ret = ompt::proxy_data_ptr(real);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int
|
||||
proxy_get_parallel_info(int ancestor_level, ompt_data_t** parallel_data, int* team_size)
|
||||
{
|
||||
ompt_data_t* tdata;
|
||||
int tteam_size;
|
||||
int ret = real_get_parallel_info(ancestor_level, &tdata, &tteam_size);
|
||||
if(ret != 2) return ret;
|
||||
if(team_size != nullptr) *team_size = tteam_size;
|
||||
if(parallel_data != nullptr) *parallel_data = ompt::proxy_data_ptr(tdata);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int
|
||||
proxy_get_task_info(int ancestor_level,
|
||||
int* flags,
|
||||
ompt_data_t** task_data,
|
||||
ompt_frame_t** task_frame,
|
||||
ompt_data_t** parallel_data,
|
||||
int* thread_num)
|
||||
{
|
||||
int tflags, tthread_num;
|
||||
ompt_data_t * ttask_data, *tparallel_data;
|
||||
ompt_frame_t* ttask_frame;
|
||||
int ret = real_get_task_info(
|
||||
ancestor_level, &tflags, &ttask_data, &ttask_frame, &tparallel_data, &tthread_num);
|
||||
if(ret != 2) return ret;
|
||||
if(flags != nullptr) *flags = tflags;
|
||||
if(task_data != nullptr) *task_data = ompt::proxy_data_ptr(ttask_data);
|
||||
if(task_frame != nullptr) *task_frame = ttask_frame;
|
||||
if(parallel_data != nullptr) *parallel_data = ompt::proxy_data_ptr(tparallel_data);
|
||||
if(thread_num != nullptr) *thread_num = tthread_num;
|
||||
return ret;
|
||||
}
|
||||
|
||||
int
|
||||
proxy_get_target_info(uint64_t* device_num, ompt_id_t* target_id, ompt_id_t* host_op_id)
|
||||
{
|
||||
uint64_t tdevice_num;
|
||||
ompt_id_t ttarget_id, thost_op_id;
|
||||
int ret = real_get_target_info(&tdevice_num, &ttarget_id, &thost_op_id);
|
||||
if(ret != 1) return ret;
|
||||
if(device_num != nullptr) *device_num = tdevice_num;
|
||||
if(target_id != nullptr) *target_id = ttarget_id;
|
||||
if(host_op_id != nullptr) *host_op_id = thost_op_id;
|
||||
return ret;
|
||||
}
|
||||
|
||||
#define SETCB(name) \
|
||||
ompt_callback_table.ompt_##name = reinterpret_cast<ompt_##name##_t>(lookup("ompt_" #name)); \
|
||||
LOG_IF(FATAL, ompt_callback_table.ompt_##name == nullptr) \
|
||||
<< "rocprofiler-sdk OMPT cannot find " \
|
||||
"ompt_" #name
|
||||
|
||||
#define PROXYCB(name) \
|
||||
SETCB(name); \
|
||||
real_##name = ompt_callback_table.ompt_##name; \
|
||||
ompt_callback_table.ompt_##name = proxy_##name
|
||||
|
||||
int
|
||||
initialize(ompt_function_lookup_t lookup, int /*initial_device_num*/, ompt_data_t* /*tool_data*/)
|
||||
{
|
||||
init_status.store(-1);
|
||||
|
||||
tool_finalize = static_cast<ompt_finalize_tool_t>(lookup("ompt_finalize_tool"));
|
||||
LOG_IF(FATAL, tool_finalize == nullptr)
|
||||
<< "rocprofiler-sdk OMPT cannot find ompt_finalize_tool";
|
||||
|
||||
set_callback = reinterpret_cast<ompt_set_callback_t>(lookup("ompt_set_callback"));
|
||||
LOG_IF(FATAL, set_callback == nullptr) << "rocprofiler-sdk OMPT cannot find ompt_set_callback";
|
||||
|
||||
SETCB(enumerate_states);
|
||||
SETCB(enumerate_mutex_impls);
|
||||
PROXYCB(get_thread_data); // <- need proxy
|
||||
SETCB(get_num_places);
|
||||
SETCB(get_place_proc_ids);
|
||||
SETCB(get_place_num);
|
||||
SETCB(get_partition_place_nums);
|
||||
SETCB(get_proc_id);
|
||||
SETCB(get_state);
|
||||
PROXYCB(get_parallel_info); // <= need proxy
|
||||
PROXYCB(get_task_info); // <= need proxy
|
||||
SETCB(get_task_memory);
|
||||
SETCB(get_num_devices);
|
||||
SETCB(get_num_procs);
|
||||
PROXYCB(get_target_info); // <= need proxy
|
||||
SETCB(get_unique_id);
|
||||
|
||||
set_ompt_callbacks();
|
||||
init_status.store(1);
|
||||
|
||||
return 1; // bizarre abberation in the OMPT spec, not 0
|
||||
}
|
||||
#undef SETCB
|
||||
|
||||
void
|
||||
finalize(ompt_data_t* /*tool_data*/)
|
||||
{
|
||||
fini_status.store(-1);
|
||||
|
||||
// do whatever for finalization
|
||||
|
||||
fini_status.store(1);
|
||||
}
|
||||
} // namespace
|
||||
|
||||
void
|
||||
finalize_ompt()
|
||||
{
|
||||
if(rocprofiler::ompt::tool_finalize) rocprofiler::ompt::tool_finalize();
|
||||
}
|
||||
} // namespace ompt
|
||||
} // namespace rocprofiler
|
||||
|
||||
extern "C" {
|
||||
|
||||
rocprofiler_status_t
|
||||
rocprofiler_ompt_is_initialized(int* status)
|
||||
{
|
||||
*status = ::rocprofiler::ompt::init_status.load();
|
||||
return ROCPROFILER_STATUS_SUCCESS;
|
||||
}
|
||||
|
||||
rocprofiler_status_t
|
||||
rocprofiler_ompt_is_finalized(int* status)
|
||||
{
|
||||
*status = ::rocprofiler::ompt::fini_status.load();
|
||||
return ROCPROFILER_STATUS_SUCCESS;
|
||||
}
|
||||
|
||||
ompt_start_tool_result_t*
|
||||
rocprofiler_ompt_start_tool(unsigned int /*omp_version*/, const char* /*runtime_version*/)
|
||||
{
|
||||
// log to clog since logging probably won't be initialized here
|
||||
auto _init_status = ::rocprofiler::ompt::init_status.load();
|
||||
if(_init_status != 0)
|
||||
{
|
||||
std::clog << "ERROR: rocprofiler-sdk OMPT backend has already been initialized: "
|
||||
<< _init_status << '\n';
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
// don't check contexts here, client tool may not be initialized
|
||||
auto* _result = ::rocprofiler::ompt::get_start_tool_result();
|
||||
|
||||
if(_result)
|
||||
{
|
||||
_result->initialize = ::rocprofiler::ompt::initialize;
|
||||
_result->finalize = ::rocprofiler::ompt::finalize;
|
||||
}
|
||||
|
||||
return _result;
|
||||
}
|
||||
|
||||
ompt_start_tool_result_t*
|
||||
ompt_start_tool(unsigned int omp_version, const char* runtime_version) ROCPROFILER_PUBLIC_API;
|
||||
|
||||
ompt_start_tool_result_t*
|
||||
ompt_start_tool(unsigned int omp_version, const char* runtime_version)
|
||||
{
|
||||
::rocprofiler::registration::initialize();
|
||||
return rocprofiler_ompt_start_tool(omp_version, runtime_version);
|
||||
}
|
||||
}
|
||||
+8
-2
@@ -22,5 +22,11 @@
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <rocprofiler-sdk/openmp/api_args.h>
|
||||
#include <rocprofiler-sdk/openmp/api_id.h>
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace ompt
|
||||
{
|
||||
void
|
||||
finalize_ompt();
|
||||
} // namespace ompt
|
||||
} // namespace rocprofiler
|
||||
@@ -0,0 +1,8 @@
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_LIB_OMPT_SOURCES ompt.cpp)
|
||||
set(ROCPROFILER_LIB_OMPT_HEADERS defines.hpp ompt.hpp utils.hpp)
|
||||
|
||||
target_sources(rocprofiler-sdk-object-library PRIVATE ${ROCPROFILER_LIB_OMPT_SOURCES})
|
||||
|
||||
add_subdirectory(details)
|
||||
@@ -0,0 +1,155 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "lib/common/defines.hpp"
|
||||
|
||||
#define OMPT_INFO_DEFINITION_0( \
|
||||
OMPT_NAME, BACKEND_OMPT_ID, OMPT_ID, OMPT_FUNC_TYPE, OMPT_ARG, OMPT_FUNC_PTR) \
|
||||
namespace rocprofiler \
|
||||
{ \
|
||||
namespace ompt \
|
||||
{ \
|
||||
template <> \
|
||||
struct ompt_info<OMPT_ID> : ompt_domain_info \
|
||||
{ \
|
||||
static constexpr auto ompt_idx = BACKEND_OMPT_ID; \
|
||||
static constexpr auto operation_idx = OMPT_ID; \
|
||||
static constexpr auto name = OMPT_NAME; \
|
||||
static constexpr auto unsupported = false; \
|
||||
\
|
||||
using domain_type = ompt_domain_info; \
|
||||
using this_type = ompt_info<operation_idx>; \
|
||||
\
|
||||
static constexpr auto callback_domain_idx = domain_type::callback_domain_idx; \
|
||||
static constexpr auto buffered_domain_idx = domain_type::buffered_domain_idx; \
|
||||
\
|
||||
using domain_type::args_type; \
|
||||
using domain_type::retval_type; \
|
||||
using domain_type::callback_data_type; \
|
||||
\
|
||||
static constexpr auto offset() \
|
||||
{ \
|
||||
return offsetof(ompt_table_lookup::type, OMPT_FUNC_PTR); \
|
||||
} \
|
||||
\
|
||||
static auto& get_table() { return ompt_table_lookup{}(); } \
|
||||
\
|
||||
template <typename TableT> \
|
||||
static auto& get_table_func(TableT& _table) \
|
||||
{ \
|
||||
if constexpr(std::is_pointer<TableT>::value) \
|
||||
{ \
|
||||
return CHECK_NOTNULL(_table)->OMPT_FUNC_PTR; \
|
||||
} \
|
||||
else \
|
||||
{ \
|
||||
return _table.OMPT_FUNC_PTR; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
static auto& get_table_func() { return get_table_func(get_table()); } \
|
||||
\
|
||||
template <typename DataT> \
|
||||
static auto& get_api_data_args(DataT& _data) \
|
||||
{ \
|
||||
return _data.OMPT_ARG; \
|
||||
} \
|
||||
\
|
||||
static std::vector<void*> as_arg_addr(callback_data_type) { return std::vector<void*>{}; } \
|
||||
\
|
||||
static std::vector<common::stringified_argument> as_arg_list(callback_data_type, int32_t) \
|
||||
{ \
|
||||
return {}; \
|
||||
} \
|
||||
}; \
|
||||
} \
|
||||
}
|
||||
|
||||
#define OMPT_INFO_DEFINITION_V( \
|
||||
OMPT_NAME, BACKEND_OMPT_ID, OMPT_ID, OMPT_FUNC_TYPE, OMPT_ARG, OMPT_FUNC_PTR, ...) \
|
||||
namespace rocprofiler \
|
||||
{ \
|
||||
namespace ompt \
|
||||
{ \
|
||||
template <> \
|
||||
struct ompt_info<OMPT_ID> : ompt_domain_info \
|
||||
{ \
|
||||
static constexpr auto ompt_idx = BACKEND_OMPT_ID; \
|
||||
static constexpr auto operation_idx = OMPT_ID; \
|
||||
static constexpr auto name = OMPT_NAME; \
|
||||
static constexpr auto unsupported = false; \
|
||||
\
|
||||
using domain_type = ompt_domain_info; \
|
||||
using this_type = ompt_info<operation_idx>; \
|
||||
\
|
||||
static constexpr auto callback_domain_idx = domain_type::callback_domain_idx; \
|
||||
static constexpr auto buffered_domain_idx = domain_type::buffered_domain_idx; \
|
||||
\
|
||||
using domain_type::args_type; \
|
||||
using domain_type::retval_type; \
|
||||
using domain_type::callback_data_type; \
|
||||
\
|
||||
static constexpr auto offset() \
|
||||
{ \
|
||||
return offsetof(ompt_table_lookup::type, OMPT_FUNC_PTR); \
|
||||
} \
|
||||
\
|
||||
static auto& get_table() { return ompt_table_lookup{}(); } \
|
||||
\
|
||||
template <typename TableT> \
|
||||
static auto& get_table_func(TableT& _table) \
|
||||
{ \
|
||||
if constexpr(std::is_pointer<TableT>::value) \
|
||||
{ \
|
||||
return CHECK_NOTNULL(_table)->OMPT_FUNC_PTR; \
|
||||
} \
|
||||
else \
|
||||
{ \
|
||||
return _table.OMPT_FUNC_PTR; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
static auto& get_table_func() { return get_table_func(get_table()); } \
|
||||
\
|
||||
template <typename DataT> \
|
||||
static auto& get_api_data_args(DataT& _data) \
|
||||
{ \
|
||||
return _data.OMPT_ARG; \
|
||||
} \
|
||||
\
|
||||
static std::vector<void*> as_arg_addr(callback_data_type trace_data) \
|
||||
{ \
|
||||
return std::vector<void*>{ \
|
||||
GET_ADDR_MEMBER_FIELDS(get_api_data_args(trace_data.args), __VA_ARGS__)}; \
|
||||
} \
|
||||
\
|
||||
static auto as_arg_list(callback_data_type trace_data, int32_t max_deref) \
|
||||
{ \
|
||||
return utils::stringize( \
|
||||
max_deref, \
|
||||
GET_NAMED_MEMBER_FIELDS(get_api_data_args(trace_data.args), __VA_ARGS__)); \
|
||||
} \
|
||||
}; \
|
||||
} \
|
||||
}
|
||||
@@ -0,0 +1,9 @@
|
||||
#
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_LIB_OMPT_DETAILS_SOURCES)
|
||||
set(ROCPROFILER_LIB_OMPT_DETAILS_HEADERS format.hpp)
|
||||
|
||||
target_sources(
|
||||
rocprofiler-sdk-object-library PRIVATE ${ROCPROFILER_LIB_OMPT_DETAILS_SOURCES}
|
||||
${ROCPROFILER_LIB_OMPT_DETAILS_HEADERS})
|
||||
@@ -0,0 +1,279 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
|
||||
#include "fmt/core.h"
|
||||
|
||||
#define ROCP_SDK_OMPT_FORMATTER(TYPE, ...) \
|
||||
template <> \
|
||||
struct formatter<TYPE> : rocprofiler::ompt::details::base_formatter \
|
||||
{ \
|
||||
template <typename Ctx> \
|
||||
auto format(const TYPE& v, Ctx& ctx) const \
|
||||
{ \
|
||||
return fmt::format_to(ctx.out(), __VA_ARGS__); \
|
||||
} \
|
||||
};
|
||||
|
||||
#define ROCP_SDK_OMPT_FORMAT_CASE_STMT(PREFIX, SUFFIX) \
|
||||
case PREFIX##_##SUFFIX: return fmt::format_to(ctx.out(), #SUFFIX)
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace ompt
|
||||
{
|
||||
namespace details
|
||||
{
|
||||
struct base_formatter
|
||||
{
|
||||
template <typename ParseContext>
|
||||
constexpr auto parse(ParseContext& ctx)
|
||||
{
|
||||
return ctx.begin();
|
||||
}
|
||||
};
|
||||
} // namespace details
|
||||
} // namespace ompt
|
||||
} // namespace rocprofiler
|
||||
|
||||
namespace fmt
|
||||
{
|
||||
template <>
|
||||
struct formatter<ompt_set_result_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_set_result_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, error);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, never);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, impossible);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, sometimes);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, sometimes_paired);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_set, always);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_thread_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_thread_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_thread, initial);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_thread, worker);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_thread, other);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt_thread, unknown);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_scope_endpoint_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_scope_endpoint_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, scope_begin);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, scope_end);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, scope_beginend);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dispatch_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_dispatch_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, dispatch_iteration);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, dispatch_section);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, dispatch_ws_loop_chunk);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, dispatch_taskloop_chunk);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, dispatch_distribute_chunk);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_sync_region_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_sync_region_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_explicit);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_implementation);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_taskwait);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_taskgroup);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_reduction);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit_workshare);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit_parallel);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, sync_region_barrier_teams);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_target_data_op_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_target_data_op_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_alloc);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_transfer_to_device);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_transfer_from_device);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_delete);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_associate);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_disassociate);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_alloc_async);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_transfer_to_device_async);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_transfer_from_device_async);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, target_data_delete_async);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_data_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_data_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(ctx.out(), "{}", v.value);
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_work_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_work_t& v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_loop);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_sections);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_single_executor);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_single_other);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_workshare);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_distribute);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_taskloop);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_scope);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_loop_static);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_loop_dynamic);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_loop_guided);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, work_loop_other);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_task_status_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_task_status_t& v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_complete);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_yield);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_cancel);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_detach);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_early_fulfill);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_late_fulfill);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, task_switch);
|
||||
ROCP_SDK_OMPT_FORMAT_CASE_STMT(ompt, taskwait_complete);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_frame_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_frame_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(
|
||||
ctx.out(),
|
||||
"{}exit_frame={}, enter_frame={}, exit_frame_flags={}, enter_frame_flags={}{}",
|
||||
'{',
|
||||
v.exit_frame,
|
||||
v.enter_frame,
|
||||
v.exit_frame_flags,
|
||||
v.enter_frame_flags,
|
||||
'}');
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dependence_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_dependence_t& v, Ctx& ctx) const
|
||||
{
|
||||
// stub
|
||||
return fmt::format_to(ctx.out(), "(dependence)");
|
||||
(void) v;
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dispatch_chunk_t> : rocprofiler::ompt::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_dispatch_chunk_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(
|
||||
ctx.out(), "{}start={}, iterations={}{}", '{', v.start, v.iterations, '}');
|
||||
}
|
||||
};
|
||||
} // namespace fmt
|
||||
|
||||
#undef ROCP_SDK_OMPT_FORMATTER
|
||||
#undef ROCP_SDK_OMPT_OSTREAM_FORMATTER
|
||||
#undef ROCP_SDK_OMPT_FORMAT_CASE_STMT
|
||||
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
@@ -0,0 +1,73 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#if defined(ROCPROFILER_LIB_ROCPROFILER_OMPT_OMPT_CPP_IMPL) && \
|
||||
ROCPROFILER_LIB_ROCPROFILER_OMPT_OMPT_CPP_IMPL == 1
|
||||
|
||||
# include "lib/common/mpl.hpp"
|
||||
# include "lib/rocprofiler-sdk/ompt/defines.hpp"
|
||||
# include "lib/rocprofiler-sdk/ompt/ompt.hpp"
|
||||
# include "lib/rocprofiler-sdk/ompt/utils.hpp"
|
||||
|
||||
# include <rocprofiler-sdk/external_correlation.h>
|
||||
# include <rocprofiler-sdk/ompt.h>
|
||||
|
||||
# include <type_traits>
|
||||
|
||||
// clang-format off
|
||||
OMPT_INFO_DEFINITION_V("omp_thread_begin", ompt_callback_thread_begin, ROCPROFILER_OMPT_ID_thread_begin, ompt_callback_thread_begin_t, thread_begin, ompt_thread_begin_fn, thread_type, thread_data)
|
||||
OMPT_INFO_DEFINITION_V("omp_thread_end", ompt_callback_thread_end, ROCPROFILER_OMPT_ID_thread_end, ompt_callback_thread_end_t, thread_end, ompt_thread_end_fn, thread_data)
|
||||
OMPT_INFO_DEFINITION_V("omp_parallel_begin", ompt_callback_parallel_begin, ROCPROFILER_OMPT_ID_parallel_begin, ompt_callback_parallel_begin_t, parallel_begin, ompt_parallel_begin_fn, encountering_task_data, encountering_task_frame, parallel_data, requested_parallelism, flags, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_parallel_end", ompt_callback_parallel_end, ROCPROFILER_OMPT_ID_parallel_end, ompt_callback_parallel_end_t, parallel_end, ompt_parallel_end_fn, parallel_data, encountering_task_data, flags, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_task_create", ompt_callback_task_create, ROCPROFILER_OMPT_ID_task_create, ompt_callback_task_create_t, task_create, ompt_task_create_fn, encountering_task_data, encountering_task_frame, new_task_data, flags, has_dependences, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_task_schedule", ompt_callback_task_schedule, ROCPROFILER_OMPT_ID_task_schedule, ompt_callback_task_schedule_t, task_schedule, ompt_task_schedule_fn, prior_task_data, prior_task_status, next_task_data)
|
||||
OMPT_INFO_DEFINITION_V("omp_implicit_task", ompt_callback_implicit_task, ROCPROFILER_OMPT_ID_implicit_task, ompt_callback_implicit_task_t, implicit_task, ompt_implicit_task_fn, endpoint, parallel_data, task_data, actual_parallelism, index, flags)
|
||||
OMPT_INFO_DEFINITION_V("omp_device_initialize", ompt_callback_device_initialize, ROCPROFILER_OMPT_ID_device_initialize, ompt_callback_device_initialize_t, device_initialize, ompt_device_initialize_fn, device_num, type, device, lookup, documentation)
|
||||
OMPT_INFO_DEFINITION_V("omp_device_finalize", ompt_callback_device_finalize, ROCPROFILER_OMPT_ID_device_finalize, ompt_callback_device_finalize_t, device_finalize, ompt_device_finalize_fn, device_num)
|
||||
OMPT_INFO_DEFINITION_V("omp_device_load", ompt_callback_device_load, ROCPROFILER_OMPT_ID_device_load, ompt_callback_device_load_t, device_load, ompt_device_load_fn, device_num, filename, offset_in_file, vma_in_file, bytes, host_addr, device_addr, module_id)
|
||||
// OMPT_INFO_DEFINITION_V("omp_device_unload", ompt_callback_device_unload, ROCPROFILER_OMPT_ID_device_unload, ompt_callback_device_unload_t, device_unload, ompt_device_unload_fn, device_num, module_id)
|
||||
OMPT_INFO_DEFINITION_V("omp_sync_region_wait", ompt_callback_sync_region_wait, ROCPROFILER_OMPT_ID_sync_region_wait, ompt_callback_sync_region_wait_t, sync_region_wait, ompt_sync_region_wait_fn, kind, endpoint, parallel_data, task_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_mutex_released", ompt_callback_mutex_released, ROCPROFILER_OMPT_ID_mutex_released, ompt_callback_mutex_released_t, mutex_released, ompt_mutex_released_fn, kind, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_dependences", ompt_callback_dependences, ROCPROFILER_OMPT_ID_dependences, ompt_callback_dependences_t, dependences, ompt_dependences_fn, task_data, deps, ndeps)
|
||||
OMPT_INFO_DEFINITION_V("omp_task_dependence", ompt_callback_task_dependence, ROCPROFILER_OMPT_ID_task_dependence, ompt_callback_task_dependence_t, task_dependence, ompt_task_dependence_fn, src_task_data, sink_task_data)
|
||||
OMPT_INFO_DEFINITION_V("omp_work", ompt_callback_work, ROCPROFILER_OMPT_ID_work, ompt_callback_work_t, work, ompt_work_fn, work_type, endpoint, parallel_data, task_data, count, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_masked", ompt_callback_masked, ROCPROFILER_OMPT_ID_masked, ompt_callback_masked_t, masked, ompt_masked_fn, endpoint, parallel_data, task_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_sync_region", ompt_callback_sync_region, ROCPROFILER_OMPT_ID_sync_region, ompt_callback_sync_region_t, sync_region, ompt_sync_region_fn, kind, endpoint, parallel_data, task_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_lock_init", ompt_callback_lock_init, ROCPROFILER_OMPT_ID_lock_init, ompt_callback_lock_init_t, lock_init, ompt_lock_init_fn, kind, hint, impl, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_lock_destroy", ompt_callback_lock_destroy, ROCPROFILER_OMPT_ID_lock_destroy, ompt_callback_lock_destroy_t, lock_destroy, ompt_lock_destroy_fn, kind, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_mutex_acquire", ompt_callback_mutex_acquire, ROCPROFILER_OMPT_ID_mutex_acquire, ompt_callback_mutex_acquire_t, mutex_acquire, ompt_mutex_acquire_fn, kind, hint, impl, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_mutex_acquired", ompt_callback_mutex_acquired, ROCPROFILER_OMPT_ID_mutex_acquired, ompt_callback_mutex_acquired_t, mutex_acquired, ompt_mutex_acquired_fn, kind, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_nest_lock", ompt_callback_nest_lock, ROCPROFILER_OMPT_ID_nest_lock, ompt_callback_nest_lock_t, nest_lock, ompt_nest_lock_fn, endpoint, wait_id, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_flush", ompt_callback_flush, ROCPROFILER_OMPT_ID_flush, ompt_callback_flush_t, flush, ompt_flush_fn, thread_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_cancel", ompt_callback_cancel, ROCPROFILER_OMPT_ID_cancel, ompt_callback_cancel_t, cancel, ompt_cancel_fn, task_data, flags, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_reduction", ompt_callback_reduction, ROCPROFILER_OMPT_ID_reduction, ompt_callback_reduction_t, reduction, ompt_reduction_fn, kind, endpoint, parallel_data, task_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_dispatch", ompt_callback_dispatch, ROCPROFILER_OMPT_ID_dispatch, ompt_callback_dispatch_t, dispatch, ompt_dispatch_fn, parallel_data, task_data, kind, instance)
|
||||
OMPT_INFO_DEFINITION_V("omp_target_emi", ompt_callback_target_emi, ROCPROFILER_OMPT_ID_target_emi, ompt_callback_target_emi_t, target_emi, ompt_target_emi_fn, kind, endpoint, device_num, task_data, target_task_data, target_data, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_target_data_op_emi", ompt_callback_target_data_op_emi, ROCPROFILER_OMPT_ID_target_data_op_emi, ompt_callback_target_data_op_emi_t, target_data_op_emi, ompt_target_data_op_emi_fn, endpoint, target_task_data, target_data, host_op_id, optype, src_address, src_device_num, dst_address, dst_device_num, bytes, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_target_submit_emi", ompt_callback_target_submit_emi, ROCPROFILER_OMPT_ID_target_submit_emi, ompt_callback_target_submit_emi_t, target_submit_emi, ompt_target_submit_emi_fn, endpoint, target_data, host_op_id, requested_num_teams)
|
||||
// OMPT_INFO_DEFINITION_V("omp_target_map_emi", ompt_callback_target_map_emi, ROCPROFILER_OMPT_ID_target_map_emi, ompt_callback_target_map_emi_t, target_map_emi, ompt_target_map_emi_fn, nitems, host_addr, device_addr, bytes, mapping_flags, codeptr_ra)
|
||||
OMPT_INFO_DEFINITION_V("omp_error", ompt_callback_error, ROCPROFILER_OMPT_ID_error, ompt_callback_error_t, error, ompt_error_fn, severity, message, length, codeptr_ra)
|
||||
// clang-format on
|
||||
|
||||
#else
|
||||
# error "Do not compile this file directly. It is included by lib/rocprofiler-sdk/ompt/ompt.cpp"
|
||||
#endif
|
||||
@@ -0,0 +1,200 @@
|
||||
// mit license
|
||||
//
|
||||
// copyright (c) 2023 advanced micro devices, inc. all rights reserved.
|
||||
//
|
||||
// permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "software"), to deal
|
||||
// in the software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the software, and to permit persons to whom the software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// the above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the software.
|
||||
//
|
||||
// the software is provided "as is", without warranty of any kind, express or
|
||||
// implied, including but not limited to the warranties of merchantability,
|
||||
// fitness for a particular purpose and noninfringement. in no event shall the
|
||||
// authors or copyright holders be liable for any claim, damages or other
|
||||
// liability, whether in an action of contract, tort or otherwise, arising from,
|
||||
// out of or in connection with the software or the use or other dealings in
|
||||
// the software.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "lib/rocprofiler-sdk/context/correlation_id.hpp"
|
||||
#include "lib/rocprofiler-sdk/tracing/fwd.hpp"
|
||||
|
||||
#include <rocprofiler-sdk/buffer_tracing.h>
|
||||
#include <rocprofiler-sdk/callback_tracing.h>
|
||||
#include <rocprofiler-sdk/external_correlation.h>
|
||||
#include <rocprofiler-sdk/fwd.h>
|
||||
#include <rocprofiler-sdk/ompt.h>
|
||||
#include <rocprofiler-sdk/ompt/api_id.h>
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
|
||||
#include <cstdint>
|
||||
#include <deque>
|
||||
#include <vector>
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace ompt
|
||||
{
|
||||
struct ompt_table
|
||||
{
|
||||
ompt_callback_thread_begin_t ompt_thread_begin_fn = nullptr;
|
||||
ompt_callback_thread_end_t ompt_thread_end_fn = nullptr;
|
||||
ompt_callback_parallel_begin_t ompt_parallel_begin_fn = nullptr;
|
||||
ompt_callback_parallel_end_t ompt_parallel_end_fn = nullptr;
|
||||
ompt_callback_task_create_t ompt_task_create_fn = nullptr;
|
||||
ompt_callback_task_schedule_t ompt_task_schedule_fn = nullptr;
|
||||
ompt_callback_implicit_task_t ompt_implicit_task_fn = nullptr;
|
||||
ompt_callback_device_initialize_t ompt_device_initialize_fn = nullptr;
|
||||
ompt_callback_device_finalize_t ompt_device_finalize_fn = nullptr;
|
||||
ompt_callback_device_load_t ompt_device_load_fn = nullptr;
|
||||
// ompt_callback_device_unload_t ompt_device_unload_fn = nullptr;
|
||||
ompt_callback_sync_region_t ompt_sync_region_wait_fn = nullptr;
|
||||
ompt_callback_mutex_t ompt_mutex_released_fn = nullptr;
|
||||
ompt_callback_dependences_t ompt_dependences_fn = nullptr;
|
||||
ompt_callback_task_dependence_t ompt_task_dependence_fn = nullptr;
|
||||
ompt_callback_work_t ompt_work_fn = nullptr;
|
||||
ompt_callback_masked_t ompt_masked_fn = nullptr;
|
||||
ompt_callback_target_map_t ompt_target_map_fn = nullptr;
|
||||
ompt_callback_sync_region_t ompt_sync_region_fn = nullptr;
|
||||
ompt_callback_mutex_acquire_t ompt_lock_init_fn = nullptr;
|
||||
ompt_callback_mutex_t ompt_lock_destroy_fn = nullptr;
|
||||
ompt_callback_mutex_acquire_t ompt_mutex_acquire_fn = nullptr;
|
||||
ompt_callback_mutex_t ompt_mutex_acquired_fn = nullptr;
|
||||
ompt_callback_nest_lock_t ompt_nest_lock_fn = nullptr;
|
||||
ompt_callback_flush_t ompt_flush_fn = nullptr;
|
||||
ompt_callback_cancel_t ompt_cancel_fn = nullptr;
|
||||
ompt_callback_sync_region_t ompt_reduction_fn = nullptr;
|
||||
ompt_callback_dispatch_t ompt_dispatch_fn = nullptr;
|
||||
ompt_callback_target_emi_t ompt_target_emi_fn = nullptr;
|
||||
ompt_callback_target_data_op_emi_t ompt_target_data_op_emi_fn = nullptr;
|
||||
ompt_callback_target_submit_emi_t ompt_target_submit_emi_fn = nullptr;
|
||||
// ompt_callback_target_map_emi_t ompt_target_map_emi_fn = nullptr;
|
||||
ompt_callback_error_t ompt_error_fn = nullptr;
|
||||
};
|
||||
|
||||
struct ompt_domain_info
|
||||
{
|
||||
using args_type = rocprofiler_ompt_args_t;
|
||||
using retval_type = void;
|
||||
using callback_data_type = rocprofiler_callback_tracing_ompt_data_t;
|
||||
using buffer_data_type = rocprofiler_buffer_tracing_ompt_record_t;
|
||||
using enum_type = rocprofiler_ompt_operation_t;
|
||||
static constexpr auto callback_domain_idx = ROCPROFILER_CALLBACK_TRACING_OMPT;
|
||||
static constexpr auto buffered_domain_idx = ROCPROFILER_BUFFER_TRACING_OMPT;
|
||||
static constexpr auto none = ROCPROFILER_OMPT_ID_NONE;
|
||||
static constexpr auto last = ROCPROFILER_OMPT_ID_LAST;
|
||||
static constexpr auto ompt_last = ROCPROFILER_OMPT_ID_callback_functions;
|
||||
static constexpr auto external_correlation_id_domain_idx =
|
||||
ROCPROFILER_EXTERNAL_CORRELATION_REQUEST_OMPT;
|
||||
};
|
||||
|
||||
using buffer_ompt_record_t = rocprofiler_buffer_tracing_ompt_record_t;
|
||||
using callback_ompt_data_t = rocprofiler_callback_tracing_ompt_data_t;
|
||||
|
||||
// save state for ompt between callbacks
|
||||
struct ompt_save_state
|
||||
{
|
||||
uint64_t thr_id; // thread this was created on
|
||||
uint64_t start_timestamp; // timestamp when it was created
|
||||
rocprofiler_ompt_operation_t operation_idx; // for error checking
|
||||
context::correlation_id* corr_id; // correlation id
|
||||
tracing::external_correlation_id_map_t external_corr_ids;
|
||||
tracing::callback_context_data_vec_t callback_contexts;
|
||||
tracing::buffered_context_data_vec_t buffered_contexts;
|
||||
};
|
||||
|
||||
// proxy for ompt_data_t pointers received from OMPT callbacks
|
||||
// SDK gets index 0, client get index 1
|
||||
class ompt_data_proxy
|
||||
{
|
||||
public:
|
||||
ompt_data_t* get_client_ptr(ompt_data_t* ompt_ptr) { return get<1>(ompt_ptr); }
|
||||
ompt_data_t* get_internal_ptr(ompt_data_t* ompt_ptr) { return get<0>(ompt_ptr); }
|
||||
|
||||
private:
|
||||
struct proxy_ptrs
|
||||
{
|
||||
std::array<ompt_data_t, 2> v;
|
||||
};
|
||||
|
||||
// get the proxy pointer for idx. If ompt_ptr->ptr is 0,
|
||||
// allocate a new proxy struct and assign it to ompt_ptr->ptr
|
||||
// return the address of the requested proxy element,
|
||||
// or null if ompt_ptr is null
|
||||
template <int idx>
|
||||
ompt_data_t* get(ompt_data_t* ompt_ptr)
|
||||
{
|
||||
static constexpr proxy_ptrs nulval{};
|
||||
if(ompt_ptr == nullptr) return nullptr;
|
||||
if(ompt_ptr->ptr == nullptr)
|
||||
{
|
||||
std::lock_guard<std::mutex> lk(m);
|
||||
ompt_ptr->ptr = &proxies.emplace_back(nulval);
|
||||
}
|
||||
auto* ptr = static_cast<proxy_ptrs*>(ompt_ptr->ptr);
|
||||
return &(ptr->v[idx]);
|
||||
}
|
||||
|
||||
std::deque<proxy_ptrs> proxies;
|
||||
std::mutex m;
|
||||
};
|
||||
|
||||
// function to return client pointer for use outside
|
||||
ompt_data_t*
|
||||
proxy_data_ptr(ompt_data_t* real_ptr);
|
||||
|
||||
struct ompt_task_save_state
|
||||
{
|
||||
context::correlation_id* corr_id;
|
||||
int task_flags;
|
||||
};
|
||||
|
||||
template <size_t opidx>
|
||||
struct ompt_impl
|
||||
{
|
||||
template <typename DataArgst, typename... Args>
|
||||
static void set_data_args(DataArgst&, Args... args);
|
||||
|
||||
template <typename... Args>
|
||||
static void begin(ompt_data_t* data, Args... args);
|
||||
|
||||
template <typename... Args>
|
||||
static void end(ompt_data_t* data, Args... args);
|
||||
|
||||
template <typename... Args>
|
||||
static context::correlation_id* event_common(Args... args);
|
||||
|
||||
template <typename... Args>
|
||||
static void event(Args&&... args);
|
||||
};
|
||||
|
||||
template <size_t OpIdx>
|
||||
struct ompt_info;
|
||||
|
||||
const char*
|
||||
name_by_id(uint32_t id);
|
||||
|
||||
std::vector<uint32_t>
|
||||
get_ids();
|
||||
|
||||
void
|
||||
iterate_args(uint32_t id,
|
||||
const rocprofiler_callback_tracing_ompt_data_t& data,
|
||||
rocprofiler_callback_tracing_operation_args_cb_t callback,
|
||||
int32_t max_deref,
|
||||
void* user_data);
|
||||
|
||||
using ompt_update_func = void (*)(const char* cbname, ompt_callback_t* cbf, int cbnum);
|
||||
void update_table(ompt_update_func);
|
||||
|
||||
void
|
||||
update_callback(rocprofiler_ompt_callback_functions_t& cb_functions);
|
||||
|
||||
} // namespace ompt
|
||||
} // namespace rocprofiler
|
||||
@@ -0,0 +1,72 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "lib/common/stringize_arg.hpp"
|
||||
#include "lib/rocprofiler-sdk/ompt/details/format.hpp"
|
||||
|
||||
#include <rocprofiler-sdk/ompt/omp-tools.h>
|
||||
#include <rocprofiler-sdk/version.h>
|
||||
|
||||
#include <fmt/core.h>
|
||||
#include <fmt/ranges.h>
|
||||
|
||||
#include <sstream>
|
||||
#include <type_traits>
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace ompt
|
||||
{
|
||||
namespace utils
|
||||
{
|
||||
template <typename Tp>
|
||||
auto
|
||||
stringize_impl(const Tp& _v)
|
||||
{
|
||||
using value_type = std::decay_t<Tp>;
|
||||
|
||||
if constexpr(fmt::is_formattable<value_type>::value && !std::is_pointer<value_type>::value)
|
||||
{
|
||||
return fmt::format("{}", _v);
|
||||
}
|
||||
else
|
||||
{
|
||||
auto _ss = std::stringstream{};
|
||||
_ss << _v;
|
||||
return _ss.str();
|
||||
}
|
||||
}
|
||||
|
||||
template <typename... Args>
|
||||
auto
|
||||
stringize(int32_t max_deref, Args... args)
|
||||
{
|
||||
using array_type = common::stringified_argument_array_t<sizeof...(Args)>;
|
||||
(void) max_deref;
|
||||
return array_type{common::stringize_arg(
|
||||
max_deref, args, [](const auto& _v) { return stringize_impl(_v); })...};
|
||||
}
|
||||
} // namespace utils
|
||||
} // namespace ompt
|
||||
} // namespace rocprofiler
|
||||
@@ -1,8 +0,0 @@
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_LIB_OPENMP_SOURCES openmp.cpp)
|
||||
set(ROCPROFILER_LIB_OPENMP_HEADERS defines.hpp openmp.hpp utils.hpp)
|
||||
|
||||
target_sources(rocprofiler-object-library PRIVATE ${ROCPROFILER_LIB_OPENMP_SOURCES})
|
||||
|
||||
add_subdirectory(details)
|
||||
@@ -1,9 +0,0 @@
|
||||
#
|
||||
#
|
||||
#
|
||||
set(ROCPROFILER_LIB_OPENMP_DETAILS_SOURCES)
|
||||
set(ROCPROFILER_LIB_OPENMP_DETAILS_HEADERS format.hpp)
|
||||
|
||||
target_sources(
|
||||
rocprofiler-object-library PRIVATE ${ROCPROFILER_LIB_OPENMP_DETAILS_SOURCES}
|
||||
${ROCPROFILER_LIB_OPENMP_DETAILS_HEADERS})
|
||||
@@ -1,279 +0,0 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <rocprofiler-sdk/openmp/omp-tools.h>
|
||||
|
||||
#include "fmt/core.h"
|
||||
|
||||
#define ROCP_SDK_OPENMP_FORMATTER(TYPE, ...) \
|
||||
template <> \
|
||||
struct formatter<TYPE> : rocprofiler::openmp::details::base_formatter \
|
||||
{ \
|
||||
template <typename Ctx> \
|
||||
auto format(const TYPE& v, Ctx& ctx) const \
|
||||
{ \
|
||||
return fmt::format_to(ctx.out(), __VA_ARGS__); \
|
||||
} \
|
||||
};
|
||||
|
||||
#define ROCP_SDK_OPENMP_FORMAT_CASE_STMT(PREFIX, SUFFIX) \
|
||||
case PREFIX##_##SUFFIX: return fmt::format_to(ctx.out(), #SUFFIX)
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace openmp
|
||||
{
|
||||
namespace details
|
||||
{
|
||||
struct base_formatter
|
||||
{
|
||||
template <typename ParseContext>
|
||||
constexpr auto parse(ParseContext& ctx)
|
||||
{
|
||||
return ctx.begin();
|
||||
}
|
||||
};
|
||||
} // namespace details
|
||||
} // namespace openmp
|
||||
} // namespace rocprofiler
|
||||
|
||||
namespace fmt
|
||||
{
|
||||
template <>
|
||||
struct formatter<ompt_set_result_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_set_result_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, error);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, never);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, impossible);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, sometimes);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, sometimes_paired);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_set, always);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_thread_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_thread_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_thread, initial);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_thread, worker);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_thread, other);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt_thread, unknown);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_scope_endpoint_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_scope_endpoint_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, scope_begin);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, scope_end);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, scope_beginend);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dispatch_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_dispatch_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, dispatch_iteration);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, dispatch_section);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, dispatch_ws_loop_chunk);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, dispatch_taskloop_chunk);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, dispatch_distribute_chunk);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_sync_region_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_sync_region_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_explicit);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_implementation);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_taskwait);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_taskgroup);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_reduction);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit_workshare);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_implicit_parallel);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, sync_region_barrier_teams);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_target_data_op_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(ompt_target_data_op_t v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_alloc);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_transfer_to_device);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_transfer_from_device);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_delete);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_associate);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_disassociate);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_alloc_async);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_transfer_to_device_async);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_transfer_from_device_async);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, target_data_delete_async);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_data_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_data_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(ctx.out(), "{}", v.value);
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_work_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_work_t& v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_loop);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_sections);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_single_executor);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_single_other);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_workshare);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_distribute);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_taskloop);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_scope);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_loop_static);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_loop_dynamic);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_loop_guided);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, work_loop_other);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_task_status_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_task_status_t& v, Ctx& ctx) const
|
||||
{
|
||||
switch(v)
|
||||
{
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_complete);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_yield);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_cancel);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_detach);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_early_fulfill);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_late_fulfill);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, task_switch);
|
||||
ROCP_SDK_OPENMP_FORMAT_CASE_STMT(ompt, taskwait_complete);
|
||||
}
|
||||
return fmt::format_to(ctx.out(), "Unknown");
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_frame_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_frame_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(
|
||||
ctx.out(),
|
||||
"{}exit_frame={}, enter_frame={}, exit_frame_flags={}, enter_frame_flags={}{}",
|
||||
'{',
|
||||
v.exit_frame,
|
||||
v.enter_frame,
|
||||
v.exit_frame_flags,
|
||||
v.enter_frame_flags,
|
||||
'}');
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dependence_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_dependence_t& v, Ctx& ctx) const
|
||||
{
|
||||
// stub
|
||||
return fmt::format_to(ctx.out(), "(dependence)");
|
||||
(void) v;
|
||||
}
|
||||
};
|
||||
|
||||
template <>
|
||||
struct formatter<ompt_dispatch_chunk_t> : rocprofiler::openmp::details::base_formatter
|
||||
{
|
||||
template <typename Ctx>
|
||||
auto format(const ompt_dispatch_chunk_t& v, Ctx& ctx) const
|
||||
{
|
||||
return fmt::format_to(
|
||||
ctx.out(), "{}start={}, iterations={}{}", '{', v.start, v.iterations, '}');
|
||||
}
|
||||
};
|
||||
} // namespace fmt
|
||||
|
||||
#undef ROCP_SDK_OPENMP_FORMATTER
|
||||
#undef ROCP_SDK_OPENMP_OSTREAM_FORMATTER
|
||||
#undef ROCP_SDK_OPENMP_FORMAT_CASE_STMT
|
||||
@@ -41,6 +41,7 @@
|
||||
#include "lib/rocprofiler-sdk/intercept_table.hpp"
|
||||
#include "lib/rocprofiler-sdk/internal_threading.hpp"
|
||||
#include "lib/rocprofiler-sdk/marker/marker.hpp"
|
||||
#include "lib/rocprofiler-sdk/ompt.hpp"
|
||||
#include "lib/rocprofiler-sdk/page_migration/page_migration.hpp"
|
||||
#include "lib/rocprofiler-sdk/pc_sampling/code_object.hpp"
|
||||
#include "lib/rocprofiler-sdk/pc_sampling/service.hpp"
|
||||
@@ -51,6 +52,7 @@
|
||||
#include <rocprofiler-sdk/fwd.h>
|
||||
#include <rocprofiler-sdk/hsa.h>
|
||||
#include <rocprofiler-sdk/marker.h>
|
||||
#include <rocprofiler-sdk/ompt.h>
|
||||
#include <rocprofiler-sdk/version.h>
|
||||
|
||||
#include <hsa/hsa_api_trace.h>
|
||||
@@ -639,6 +641,7 @@ finalize()
|
||||
counters::device_counting_service_finalize();
|
||||
hsa::queue_controller_fini();
|
||||
thread_trace::finalize();
|
||||
ompt::finalize_ompt();
|
||||
page_migration::finalize();
|
||||
#if ROCPROFILER_SDK_HSA_PC_SAMPLING > 0
|
||||
// WARNING: this must precede `code_object::finalize()`
|
||||
|
||||
@@ -11,6 +11,10 @@ race:libamdhip64.so
|
||||
# data race arising from hsa runtime
|
||||
race:libhsa-runtime64.so
|
||||
|
||||
# data race(s) arising from OpenMP
|
||||
race:__kmp_resume_template
|
||||
race:__kmp_suspend_initialize_thread
|
||||
|
||||
# unlock of an unlocked mutex (or by a wrong thread)
|
||||
mutex:libhsa-runtime64.so
|
||||
|
||||
|
||||
Ссылка в новой задаче
Block a user