Rearrange cpp cmake package

Author: myl7

CMakeLists.txt

@@ -25,11 +25,12 @@ include_directories(${PYTHON_INCLUDE_DIRS})
 
 add_library(
   fssprgcuda SHARED
-  lib.cpp
-  torchcsprng/kernels.cu
-  torchcsprng/owcf.cu
-  torchcsprng/aes.cu
+  src/fssprgcuda.cpp
+  src/torchcsprng/kernels.cu
+  src/torchcsprng/owcf.cu
+  src/torchcsprng/aes.cu
 )
 set_target_properties(fssprgcuda PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
-target_link_libraries(fssprgcuda "${TORCH_LIBRARIES}")
+target_include_directories(fssprgcuda PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/include")
+target_link_libraries(fssprgcuda PRIVATE "${TORCH_LIBRARIES}")
 target_compile_options(fssprgcuda PRIVATE $<$<COMPILE_LANGUAGE: CUDA>: --expt-extended-lambda>)

include/fssprgcuda.h

@@ -0,0 +1,12 @@
+// Copyright (C) myl7
+// SPDX-License-Identifier: BSD-3-Clause
+
+#pragma once
+
+#include <cstdint>
+
+namespace fssprgcuda {
+
+void matyas_meyer_oseas_aes128(uint8_t *buf, int64_t buf_size, const uint8_t *key);
+
+}  // namespace fssprgcuda

lib.h

@@ -1,16 +1,18 @@
 // Copyright (C) myl7
 // SPDX-License-Identifier: BSD-3-Clause
 
-#include "lib.h"
+#include "fssprgcuda.h"
 #include <torch/torch.h>
+#include "torchcsprng/kernels.cuh"
 
 using torch::Tensor;
-
-extern Tensor encrypt(Tensor buf, const uint8_t *key, size_t key_size, const std::string &cipher);
+using torch::csprng::cuda::encrypt;
 
 constexpr size_t block_t_size = 16;
 
-void csprng_matyas_meyer_oseas_aes128(uint8_t *buf, size_t buf_size, const uint8_t *key, size_t key_size) {
+namespace fssprgcuda {
+
+void matyas_meyer_oseas_aes128(uint8_t *buf, size_t buf_size, const uint8_t *key, size_t key_size) {
   const auto input_size_bytes = buf_size;
   TORCH_CHECK(input_size_bytes % block_t_size == 0, "input size in bytes(", input_size_bytes,
     ") is not a multiple of block size(", block_t_size, ")");
@@ -21,3 +23,5 @@ void csprng_matyas_meyer_oseas_aes128(uint8_t *buf, size_t buf_size, const uint8
 
   const auto output = encrypt(input, key, key_size, "aes128");
 }
+
+}  // namespace fssprgcuda