initial commit

Author: js1010

.gitmodules

@@ -0,0 +1,6 @@
+[submodule "3rd/json11"]
+	path = 3rd/json11
+	url = https://github.com/dropbox/json11
+[submodule "3rd/spdlog"]
+	path = 3rd/spdlog
+	url = https://github.com/gabime/spdlog

3rd/json11

@@ -0,0 +1,43 @@
+// Copyright (c) 2021 Jisang Yoon
+// All rights reserved.
+//
+// This source code is licensed under the Apache 2.0 license found in the
+// LICENSE file in the root directory of this source tree.
+#pragma once
+#include <thrust/copy.h>
+#include <thrust/fill.h>
+#include <thrust/random.h>
+#include <thrust/host_vector.h>
+#include <thrust/device_vector.h>
+#include <thrust/binary_search.h>
+#include <thrust/execution_policy.h>
+
+#include <omp.h>
+#include <set>
+#include <random>
+#include <memory>
+#include <string>
+#include <fstream>
+#include <utility>
+#include <queue>
+#include <deque>
+#include <functional>
+#include <vector>
+#include <cmath>
+#include <chrono> // NOLINT
+
+#include "json11.hpp"
+#include "log.hpp"
+#include "types.hpp"
+
+namespace culda {
+
+class CuLDA {
+ public: 
+  CuLDA();
+  ~CuLDA();
+ private:
+  thrust::device_vector<cuda_scalar> device_data_;
+};
+
+} // namespace culda

3rd/spdlog

@@ -0,0 +1,58 @@
+// Copyright (c) 2020 Jisang Yoon
+// All rights reserved.
+//
+// This source code is licensed under the Apache 2.0 license found in the
+// LICENSE file in the root directory of this source tree.
+#pragma once
+#include <cuda_fp16.h>
+
+// experimental codes to use half precision
+// not properly working yet..
+// #define HALF_PRECISION 1
+
+// #if __CUDA_ARCH__ < 530
+//   #undef HALF_PRECISION
+// #endif
+
+#ifdef HALF_PRECISION
+  typedef half cuda_scalar;
+  #define mul(x, y) ( __hmul(x, y) )
+  #define add(x, y) ( __hadd(x, y) )
+  #define sub(x, y) ( __hsub(x, y) )
+  #define gt(x, y) ( __hgt(x, y) )  // x > y
+  #define ge(x, y) ( __hge(x, y) )  // x >= y
+  #define lt(x, y) ( __hlt(x, y) )  // x < y
+  #define le(x, y) ( __hle(x, y) )  // x <= y
+  #define out_scalar(x) ( __half2float(x) )
+  #define conversion(x) ( __float2half(x) )
+#else
+  typedef float cuda_scalar;
+  #define mul(x, y) ( x * y )
+  #define add(x, y) ( x + y )
+  #define sub(x, y) ( x - y )
+  #define gt(x, y) ( x > y )
+  #define ge(x, y) ( x >= y )
+  #define lt(x, y) ( x < y )
+  #define le(x, y) ( x <= y )
+  #define out_scalar(x) ( x )
+  #define conversion(x) ( x )
+#endif
+
+#define WARP_SIZE 32
+
+struct Neighbor {
+  cuda_scalar distance;
+  int nodeid;
+  bool checked;
+};
+
+// to manage the compatibility with hnswlib
+typedef unsigned int tableint;
+typedef unsigned int sizeint;
+typedef float scalar;
+typedef size_t labeltype;
+
+enum DIST_TYPE {
+  DOT,
+  L2,
+};