xref: /oneTBB/src/tbb/small_object_pool_impl.h (revision b15aabb3)
151c0b2f7Stbbdev /*
2*b15aabb3Stbbdev     Copyright (c) 2020-2021 Intel Corporation
351c0b2f7Stbbdev 
451c0b2f7Stbbdev     Licensed under the Apache License, Version 2.0 (the "License");
551c0b2f7Stbbdev     you may not use this file except in compliance with the License.
651c0b2f7Stbbdev     You may obtain a copy of the License at
751c0b2f7Stbbdev 
851c0b2f7Stbbdev         http://www.apache.org/licenses/LICENSE-2.0
951c0b2f7Stbbdev 
1051c0b2f7Stbbdev     Unless required by applicable law or agreed to in writing, software
1151c0b2f7Stbbdev     distributed under the License is distributed on an "AS IS" BASIS,
1251c0b2f7Stbbdev     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1351c0b2f7Stbbdev     See the License for the specific language governing permissions and
1451c0b2f7Stbbdev     limitations under the License.
1551c0b2f7Stbbdev */
1651c0b2f7Stbbdev 
1751c0b2f7Stbbdev #ifndef __TBB_small_object_pool_impl_H
1851c0b2f7Stbbdev #define __TBB_small_object_pool_impl_H
1951c0b2f7Stbbdev 
2049e08aacStbbdev #include "oneapi/tbb/detail/_small_object_pool.h"
2149e08aacStbbdev #include "oneapi/tbb/detail/_utils.h"
2251c0b2f7Stbbdev 
2351c0b2f7Stbbdev #include <cstddef>
2451c0b2f7Stbbdev #include <cstdint>
2551c0b2f7Stbbdev #include <atomic>
2651c0b2f7Stbbdev 
2751c0b2f7Stbbdev 
2851c0b2f7Stbbdev namespace tbb {
2951c0b2f7Stbbdev namespace detail {
3051c0b2f7Stbbdev namespace r1 {
3151c0b2f7Stbbdev 
3251c0b2f7Stbbdev class thread_data;
3351c0b2f7Stbbdev 
3451c0b2f7Stbbdev class small_object_pool_impl : public d1::small_object_pool
3551c0b2f7Stbbdev {
3651c0b2f7Stbbdev     static constexpr std::size_t small_object_size = 256;
3751c0b2f7Stbbdev     struct small_object {
3851c0b2f7Stbbdev         small_object* next;
3951c0b2f7Stbbdev     };
4051c0b2f7Stbbdev     static small_object* const dead_public_list;
4151c0b2f7Stbbdev public:
4251c0b2f7Stbbdev     void* allocate_impl(small_object_pool*& allocator, std::size_t number_of_bytes);
4351c0b2f7Stbbdev     void deallocate_impl(void* ptr, std::size_t number_of_bytes, thread_data& td);
4451c0b2f7Stbbdev     void destroy();
4551c0b2f7Stbbdev private:
4651c0b2f7Stbbdev     static std::int64_t cleanup_list(small_object* list);
4751c0b2f7Stbbdev     ~small_object_pool_impl() = default;
4851c0b2f7Stbbdev private:
alignas(max_nfs_size)4951c0b2f7Stbbdev     alignas(max_nfs_size) small_object* m_private_list;
5051c0b2f7Stbbdev     std::int64_t m_private_counter{};
5151c0b2f7Stbbdev     alignas(max_nfs_size) std::atomic<small_object*> m_public_list;
5251c0b2f7Stbbdev     std::atomic<std::int64_t> m_public_counter{};
5351c0b2f7Stbbdev };
5451c0b2f7Stbbdev 
5551c0b2f7Stbbdev } // namespace r1
5651c0b2f7Stbbdev } // namespace detail
5751c0b2f7Stbbdev } // namespace tbb
5851c0b2f7Stbbdev 
5951c0b2f7Stbbdev #endif /* __TBB_small_object_pool_impl_H */
60