feat(cmake): Support cuda build

Signed-off-by: KunoiSayami <[email protected]>

3 files changed, 9 insertions, 6 deletions

diff --git a/db/memtable.h b/db/memtable.h
index 9d986b1..3d3444b 100644
--- a/db/memtable.h
+++ b/db/memtable.h
@@ -5,12 +5,13 @@
 #ifndef STORAGE_LEVELDB_DB_MEMTABLE_H_
 #define STORAGE_LEVELDB_DB_MEMTABLE_H_
 
-#include <string>
-
 #include "db/dbformat.h"
 #include "db/skiplist.h"
+#include <string>
+
 #include "leveldb/db.h"
-#include "util/arena.h"
+
+#include "util/arena.cuh"
 
 namespace leveldb {
 
diff --git a/db/skiplist.h b/db/skiplist.h
index a59b45b..13df778 100644
--- a/db/skiplist.h
+++ b/db/skiplist.h
@@ -31,7 +31,7 @@
 #include <cassert>
 #include <cstdlib>
 
-#include "util/arena.h"
+#include "util/arena.cuh"
 #include "util/random.h"
 
 namespace leveldb {
diff --git a/db/skiplist_test.cc b/db/skiplist_test.cc
index 79a5b86..402cb47 100644
--- a/db/skiplist_test.cc
+++ b/db/skiplist_test.cc
@@ -7,15 +7,17 @@
 #include <atomic>
 #include <set>
 
-#include "gtest/gtest.h"
 #include "leveldb/env.h"
+
 #include "port/port.h"
 #include "port/thread_annotations.h"
-#include "util/arena.h"
+#include "util/arena.cuh"
 #include "util/hash.h"
 #include "util/random.h"
 #include "util/testutil.h"
 
+#include "gtest/gtest.h"
+
 namespace leveldb {
 
 typedef uint64_t Key;