diff options
author | Nicolas Geoffray <ngeoffray@google.com> | 2014-02-18 16:43:35 +0000 |
---|---|---|
committer | Nicolas Geoffray <ngeoffray@google.com> | 2014-02-19 16:48:02 +0000 |
commit | 68a5fefa90f03fdf5a238ac85c9439c6b03eae96 (patch) | |
tree | 0f3ba23e5d3e2bbd211023fd7ce0740f694e7215 /compiler/utils/arena_allocator.h | |
parent | 0bf2ed98bc7d529a28ab470d36308ee2358f5a1c (diff) | |
download | art-68a5fefa90f03fdf5a238ac85c9439c6b03eae96.zip art-68a5fefa90f03fdf5a238ac85c9439c6b03eae96.tar.gz art-68a5fefa90f03fdf5a238ac85c9439c6b03eae96.tar.bz2 |
Initial check-in of an optimizing compiler.
The classes and the names are very much inspired by V8/Dart.
It currently only supports the RETURN_VOID dex instruction,
and there is a pretty printer to check if the building of the
graph is correct.
Change-Id: Id5ef1b317ab997010d4e3888e456c26bef1ab9c0
Diffstat (limited to 'compiler/utils/arena_allocator.h')
-rw-r--r-- | compiler/utils/arena_allocator.h | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/compiler/utils/arena_allocator.h b/compiler/utils/arena_allocator.h new file mode 100644 index 0000000..56cedfe --- /dev/null +++ b/compiler/utils/arena_allocator.h @@ -0,0 +1,157 @@ +/* + * Copyright (C) 2013 The Android Open Source Project + * + * 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. + */ + +#ifndef ART_COMPILER_UTILS_ARENA_ALLOCATOR_H_ +#define ART_COMPILER_UTILS_ARENA_ALLOCATOR_H_ + +#include <stdint.h> +#include <stddef.h> + +#include "base/mutex.h" +#include "mem_map.h" + +namespace art { + +class Arena; +class ArenaPool; +class ArenaAllocator; + +class Arena { + public: + static constexpr size_t kDefaultSize = 128 * KB; + explicit Arena(size_t size = kDefaultSize); + ~Arena(); + void Reset(); + uint8_t* Begin() { + return memory_; + } + + uint8_t* End() { + return memory_ + size_; + } + + size_t Size() const { + return size_; + } + + size_t RemainingSpace() const { + return Size() - bytes_allocated_; + } + + private: + size_t bytes_allocated_; + uint8_t* memory_; + size_t size_; + MemMap* map_; + Arena* next_; + friend class ArenaPool; + friend class ArenaAllocator; + DISALLOW_COPY_AND_ASSIGN(Arena); +}; + +class ArenaPool { + public: + ArenaPool(); + ~ArenaPool(); + Arena* AllocArena(size_t size); + void FreeArena(Arena* arena); + + private: + Mutex lock_ DEFAULT_MUTEX_ACQUIRED_AFTER; + Arena* free_arenas_ GUARDED_BY(lock_); + DISALLOW_COPY_AND_ASSIGN(ArenaPool); +}; + +class ArenaAllocator { + public: + // Type of allocation for memory tuning. + enum ArenaAllocKind { + kAllocMisc, + kAllocBB, + kAllocLIR, + kAllocMIR, + kAllocDFInfo, + kAllocGrowableArray, + kAllocGrowableBitMap, + kAllocDalvikToSSAMap, + kAllocDebugInfo, + kAllocSuccessor, + kAllocRegAlloc, + kAllocData, + kAllocPredecessors, + kNumAllocKinds + }; + + static constexpr bool kCountAllocations = false; + + explicit ArenaAllocator(ArenaPool* pool); + ~ArenaAllocator(); + + // Returns zeroed memory. + void* Alloc(size_t bytes, ArenaAllocKind kind) ALWAYS_INLINE { + if (UNLIKELY(running_on_valgrind_)) { + return AllocValgrind(bytes, kind); + } + bytes = (bytes + 3) & ~3; + if (UNLIKELY(ptr_ + bytes > end_)) { + // Obtain a new block. + ObtainNewArenaForAllocation(bytes); + if (UNLIKELY(ptr_ == nullptr)) { + return nullptr; + } + } + if (kCountAllocations) { + alloc_stats_[kind] += bytes; + ++num_allocations_; + } + uint8_t* ret = ptr_; + ptr_ += bytes; + return ret; + } + + void* AllocValgrind(size_t bytes, ArenaAllocKind kind); + void ObtainNewArenaForAllocation(size_t allocation_size); + size_t BytesAllocated() const; + void DumpMemStats(std::ostream& os) const; + + private: + void UpdateBytesAllocated(); + + ArenaPool* pool_; + uint8_t* begin_; + uint8_t* end_; + uint8_t* ptr_; + Arena* arena_head_; + size_t num_allocations_; + size_t alloc_stats_[kNumAllocKinds]; // Bytes used by various allocation kinds. + bool running_on_valgrind_; + + DISALLOW_COPY_AND_ASSIGN(ArenaAllocator); +}; // ArenaAllocator + +struct MemStats { + public: + void Dump(std::ostream& os) const { + arena_.DumpMemStats(os); + } + explicit MemStats(const ArenaAllocator &arena) : arena_(arena) {} + private: + const ArenaAllocator &arena_; +}; // MemStats + +} // namespace art + +#endif // ART_COMPILER_UTILS_ARENA_ALLOCATOR_H_ |