From ff62cbaad13d4bd309de5d3dda8c7051325d5663 Mon Sep 17 00:00:00 2001
From: KunoiSayami <i@leanhe.dev>
Date: Thu, 11 Nov 2021 22:01:31 +0800
Subject: feat(cmake): Support cuda build

Signed-off-by: KunoiSayami <i@leanhe.dev>
---
 db/memtable.h       | 7 ++++---
 db/skiplist.h       | 2 +-
 db/skiplist_test.cc | 6 ++++--
 3 files changed, 9 insertions(+), 6 deletions(-)

(limited to 'db')

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