From 706cae9f460fe698857f47aaa42eb517ea2cb09f Mon Sep 17 00:00:00 2001
From: KunoiSayami <i@leanhe.dev>
Date: Thu, 7 Apr 2022 01:32:17 +0800
Subject: feat: Finish implement cuda memtable

Signed-off-by: KunoiSayami <i@leanhe.dev>
---
 db/skiplist_test.cu | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

(limited to 'db/skiplist_test.cu')

diff --git a/db/skiplist_test.cu b/db/skiplist_test.cu
index e5a1256..f77fc60 100644
--- a/db/skiplist_test.cu
+++ b/db/skiplist_test.cu
@@ -441,13 +441,13 @@ __global__ void testKeysIsEqualLists(SkipList<Key, Comparator> * skiplist, const
   assert(!iter.Valid());
 }
 
-__global__ void initSkipList(Arena ** pArena, SkipList<Key, Comparator> ** pSkipList) {
+__global__ void initSkipList(ArenaCuda ** pArena, SkipList<Key, Comparator> ** pSkipList) {
   Comparator cmp;
-  *pArena = new Arena();
+  *pArena = new ArenaCuda();
   *pSkipList = new SkipList<Key, Comparator>(cmp, *pArena);
 }
 
-__global__ void freeSkipList(Arena *** pArena, SkipList<Key, Comparator> *** pSkipList) {
+__global__ void freeSkipList(ArenaCuda *** pArena, SkipList<Key, Comparator> *** pSkipList) {
     cudaFree(**pArena);
     cudaFree(**pSkipList);
     cudaFree(*pArena);
@@ -456,7 +456,7 @@ __global__ void freeSkipList(Arena *** pArena, SkipList<Key, Comparator> *** pSk
 
 TEST(SkipTest, TestInitSkiplist) {
 
-  Arena ** pArena;
+  ArenaCuda ** pArena;
   SkipList<Key, Comparator> ** pSkipList;
 
   cudaMallocManaged((void**)&pArena, sizeof(void*));
@@ -495,7 +495,7 @@ TEST(SkipTest, TestSingleCudaInsert) {
     keys[i] = .Next();
   }*/
   Key * keys = new Key[SKIPLIST_TEST_SIZE], * sorted_keys = new Key[SKIPLIST_TEST_SIZE];
-  Arena ** pArena;
+  ArenaCuda ** pArena;
   SkipList<Key, Comparator> ** skipList;
   std::set<Key> k;
 
@@ -561,7 +561,7 @@ TEST(SkipTest, TestSingleCudaInsert) {
 
 TEST(SkipTest, TestMultiThreadInsert) {
   Key * keys = new Key[SKIPLIST_TEST_SIZE], * sorted_keys = new Key[SKIPLIST_TEST_SIZE];
-  Arena ** pArena;
+  ArenaCuda ** pArena;
   SkipList<Key, Comparator> ** pSkipList;
   std::set<Key> k;
   Key * device_keys = nullptr;
-- 
cgit v1.3.1