Saran Tunyasuvunakool
2f14a7c102
Symbolization and string comparison results are now cached. PiperOrigin-RevId: 564697198 Change-Id: I0447f283a90143e2f67ecd90d895a36a078b3981
100 lines
2.8 KiB
C++
100 lines
2.8 KiB
C++
// Copyright 2022 DeepMind Technologies Limited
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "engine/engine_crossplatform.h" // IWYU pragma: keep
|
|
|
|
#if defined(__APPLE__) && defined(__AVX__)
|
|
#include <sys/sysctl.h>
|
|
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
|
|
namespace {
|
|
__attribute__((weak, visibility("default")))
|
|
extern "C" void _mj_rosettaError(const char* msg) {
|
|
fprintf(stderr, "%s\n", msg);
|
|
__asm__ __volatile__ ("ud2"); // raises SIGILL but leave this function at the top of the stack
|
|
}
|
|
|
|
__attribute__((constructor(10000), target("no-avx")))
|
|
void CheckRosetta() {
|
|
int is_translated = 0;
|
|
{
|
|
size_t len = sizeof(is_translated);
|
|
if (sysctlbyname("sysctl.proc_translated", &is_translated, &len, NULL, 0)) {
|
|
is_translated = 0;
|
|
}
|
|
}
|
|
if (is_translated) {
|
|
_mj_rosettaError("MuJoCo cannot be run under Rosetta 2 on an Apple Silicon machine.");
|
|
}
|
|
}
|
|
} // namespace
|
|
#endif // defined(__APPLE__) && defined(__AVX__)
|
|
|
|
#ifdef ADDRESS_SANITIZER
|
|
#include <sanitizer/common_interface_defs.h>
|
|
|
|
#include <array>
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <shared_mutex>
|
|
#include <string>
|
|
#include <string_view>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
|
|
namespace {
|
|
std::string_view SymbolizeCached(void* pc) {
|
|
static auto* mu = new std::shared_mutex;
|
|
static auto* pc_to_func_name_map = new std::unordered_map<void*, std::string>;
|
|
|
|
{
|
|
std::shared_lock lock(*mu);
|
|
auto it = pc_to_func_name_map->find(pc);
|
|
if (it != pc_to_func_name_map->end()) {
|
|
return it->second;
|
|
}
|
|
}
|
|
|
|
std::array<char, 256> buf;
|
|
__sanitizer_symbolize_pc(pc, "%f", buf.data(), buf.size());
|
|
{
|
|
std::unique_lock lock(*mu);
|
|
return pc_to_func_name_map->emplace(pc, buf.data()).first->second;
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
int _mj_comparePcFuncName(void* pc1, void* pc2) {
|
|
static auto* mu = new std::shared_mutex;
|
|
static auto* same_func_map = new std::map<std::pair<void*, void*>, bool>;
|
|
|
|
auto pc_pair = std::make_pair(pc1, pc2);
|
|
{
|
|
std::shared_lock lock(*mu);
|
|
auto it = same_func_map->find(pc_pair);
|
|
if (it != same_func_map->end()) {
|
|
return it->second;
|
|
}
|
|
}
|
|
|
|
bool is_same = (SymbolizeCached(pc1) == SymbolizeCached(pc2));
|
|
{
|
|
std::unique_lock lock(*mu);
|
|
return same_func_map->emplace(pc_pair, is_same).first->second;
|
|
}
|
|
}
|
|
#endif // ADDRESS_SANITIZER
|