文件
rocm-systems/source/lib/rocprof-sys/api.cpp
T
Sohaib Nadeem 0e535daa93 Initialization fixes (#154)
- Remove tooling initialization from rocprofiler_configure:
when rocprofiler configure is called from __hip_module_ctor
(which in turn is called as a global constructor when loading shared
libraries or before main in a hip program), initializing tooling
in it can cause problems because it is too early to do some of the tasks
that it involves (e.g. opening shared libraries, creating threads).
Instead, we rely on rocprofsys_main to initialize tooling later.

- Skip rocprofiler_configure if ROCPROFSYS_PRELOAD is not set since
preload is required for tooling (such as perfetto, which is used by
the rocprofiler callbacks) to be initialized.

- Revert RCCL initialization changes: These are no longer needed since rocprofsys_init_tooling_hidden will not
be called from rocprofiler_configure

- Force rocprofiler_configure in rocprofsys_init_tooling_hidden if it hasn't been
called through __hip_module_ctor global constructor
2025-04-21 17:04:24 -04:00

170 行
4.5 KiB
C++

// MIT License
//
// Copyright (c) 2022-2025 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 "api.hpp"
#include "core/debug.hpp"
#include <exception>
#include <stdexcept>
extern "C" void
rocprofsys_push_trace(const char* _name)
{
rocprofsys_push_trace_hidden(_name);
}
extern "C" void
rocprofsys_pop_trace(const char* _name)
{
rocprofsys_pop_trace_hidden(_name);
}
extern "C" int
rocprofsys_push_region(const char* _name)
{
try
{
rocprofsys_push_region_hidden(_name);
} catch(std::exception& _e)
{
ROCPROFSYS_WARNING_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
rocprofsys_pop_region(const char* _name)
{
try
{
rocprofsys_pop_region_hidden(_name);
} catch(std::exception& _e)
{
ROCPROFSYS_WARNING_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
rocprofsys_push_category_region(rocprofsys_category_t _category, const char* _name,
rocprofsys_annotation_t* _annotations,
size_t _annotation_count)
{
try
{
rocprofsys_push_category_region_hidden(_category, _name, _annotations,
_annotation_count);
} catch(std::exception& _e)
{
ROCPROFSYS_WARNING_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
rocprofsys_pop_category_region(rocprofsys_category_t _category, const char* _name,
rocprofsys_annotation_t* _annotations,
size_t _annotation_count)
{
try
{
rocprofsys_pop_category_region_hidden(_category, _name, _annotations,
_annotation_count);
} catch(std::exception& _e)
{
ROCPROFSYS_WARNING_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" void
rocprofsys_progress(const char* _name)
{
rocprofsys_progress_hidden(_name);
}
extern "C" void
rocprofsys_annotated_progress(const char* _name, rocprofsys_annotation_t* _annotations,
size_t _annotation_count)
{
rocprofsys_annotated_progress_hidden(_name, _annotations, _annotation_count);
}
extern "C" void
rocprofsys_init_library(void)
{
rocprofsys_init_library_hidden();
}
extern "C" void
rocprofsys_init_tooling(void)
{
rocprofsys_init_tooling_hidden();
}
extern "C" void
rocprofsys_init(const char* _mode, bool _rewrite, const char* _arg0)
{
rocprofsys_init_hidden(_mode, _rewrite, _arg0);
}
extern "C" void
rocprofsys_finalize(void)
{
rocprofsys_finalize_hidden();
}
extern "C" void
rocprofsys_reset_preload(void)
{
rocprofsys_reset_preload_hidden();
}
extern "C" void
rocprofsys_set_env(const char* env_name, const char* env_val)
{
rocprofsys_set_env_hidden(env_name, env_val);
}
extern "C" void
rocprofsys_set_mpi(bool use, bool attached)
{
rocprofsys_set_mpi_hidden(use, attached);
}
extern "C" void
rocprofsys_register_source(const char* file, const char* func, size_t line,
size_t address, const char* source)
{
rocprofsys_register_source_hidden(file, func, line, address, source);
}
extern "C" void
rocprofsys_register_coverage(const char* file, const char* func, size_t address)
{
rocprofsys_register_coverage_hidden(file, func, address);
}