diff options
Diffstat (limited to 'util/arena.cuh')
| -rw-r--r-- | util/arena.cuh | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/util/arena.cuh b/util/arena.cuh index f9ccb1e..b70dcb9 100644 --- a/util/arena.cuh +++ b/util/arena.cuh @@ -42,7 +42,7 @@ class Arena { struct ArenaNode { char * block; - ArenaNode * next; + cuda::atomic<ArenaNode*> next; }; // Allocation state @@ -53,8 +53,8 @@ class Arena { //thrust::host_vector<char *> blocks_; //std::vector<char*> blocks_; - ArenaNode * head_; - ArenaNode * blocks_; + cuda::atomic<ArenaNode *> head_; + cuda::atomic<ArenaNode *> blocks_; // Total memory usage of the arena. |