vecvec: expose key type, add cuda compat util

felixguendling · Jan 14, 2025 · b4b8bc7 · b4b8bc7
1 parent 577c81f
commit b4b8bc7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/include/cista/containers/vecvec.h b/include/cista/containers/vecvec.h
@@ -12,6 +12,7 @@ namespace cista {
 
 template <typename Key, typename DataVec, typename IndexVec>
 struct basic_vecvec {
+  using key = Key;
   using data_value_type = typename DataVec::value_type;
   using index_value_type = typename IndexVec::value_type;
 

diff --git a/include/cista/cuda_check.h b/include/cista/cuda_check.h
@@ -0,0 +1,7 @@
+#pragma once
+
+#if defined(__CUDA_ARCH__)
+#define CISTA_CUDA_COMPAT __host__ __device__
+#else
+#define CISTA_CUDA_COMPAT
+#endif