1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
//===--- TaskAlloc.cpp - Task-local stack allocator -----------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// A task-local allocator that obeys a stack discipline.
//
// Because allocation is task-local, and there's at most one thread
// running a task at once, no synchronization is required.
//
//===----------------------------------------------------------------------===//
#include "Runtime/Concurrency.h"
#include "Concurrency/Task.h"
#include "Concurrency/TaskPrivate.h"
using namespace swift;
namespace {
struct GlobalAllocator {
TaskAllocator allocator;
void *spaceForFirstSlab[64];
GlobalAllocator() : allocator(spaceForFirstSlab, sizeof(spaceForFirstSlab)) {}
};
} // namespace
static TaskAllocator &allocator(AsyncTask *task) {
if (task)
return task->Private.get().Allocator;
// FIXME: this fall-back shouldn't be necessary, but it's useful
// for now, since the current execution tests aren't setting up a task
// properly.
// https://github.com/apple/swift/issues/62761
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wexit-time-destructors"
static GlobalAllocator global;
#pragma clang diagnostic pop
return global.allocator;
}
void *swift::_swift_task_alloc_specific(AsyncTask *task, size_t size) {
return allocator(task).alloc(size);
}
void swift::_swift_task_dealloc_specific(AsyncTask *task, void *ptr) {
allocator(task).dealloc(ptr);
}
|
