Parallel.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2026 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <atomic>
#include <cstddef>
 
namespace open3d {
namespace utility {
 
template <typename T>
inline void AtomicAdd(std::atomic<T>& total, const T& val) noexcept {
#ifdef __cpp_lib_atomic_float
    total.fetch_add(val);
#else
    T expected = total.load(std::memory_order_relaxed);
    while (!total.compare_exchange_weak(expected, expected + val,
                                        std::memory_order_relaxed,
                                        std::memory_order_relaxed)) {
    }
#endif
}
 
int EstimateMaxThreads();
 
std::size_t& DefaultGrainSizeTBB() noexcept;
 
std::size_t& DefaultGrainSizeTBB2D() noexcept;
 
void SetMaxThreads(int num_threads);
 
}  // namespace utility
}  // namespace open3d