From 1e4c8faa097c5423604e343bf20745f57c81f2a9 Mon Sep 17 00:00:00 2001
From: KunoiSayami <i@leanhe.dev>
Date: Wed, 26 Jan 2022 02:29:32 +0800
Subject: test(skiplist): Use atomic arena

Signed-off-by: KunoiSayami <i@leanhe.dev>
---
 util/arena.cuh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'util/arena.cuh')

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.
-- 
cgit v1.3.1