v0.2.1 (#116)

* Fix a header installation bug
* Check the Python version during build
* Resolve compilation warnings

Author: chhwang

CITATION.cff

@@ -1,6 +1,6 @@
 cff-version: 1.2.0
 title: "ARK: A GPU-driven system framework for scalable AI applications"
-version: 0.2.0
+version: 0.2.1
 message: >-
   If you use this project in your research, please cite it as below.
 authors:

CMakeLists.txt

@@ -3,7 +3,7 @@
 
 set(ARK_MAJOR "0")
 set(ARK_MINOR "2")
-set(ARK_PATCH "0")
+set(ARK_PATCH "1")
 
 set(ARK_VERSION "${ARK_MAJOR}.${ARK_MINOR}.${ARK_PATCH}")
 set(ARK_SOVERSION "${ARK_MAJOR}.${ARK_MINOR}")
@@ -45,11 +45,11 @@ if(CUDAToolkit_FOUND)
     endif()
 
     if(CUDAToolkit_VERSION_MAJOR GREATER_EQUAL 11)
-        set(CMAKE_CUDA_ARCHITECTURES 70 80)
+        set(CMAKE_CUDA_ARCHITECTURES 60 70 80)
     endif()
 
     if(CUDAToolkit_VERSION_MAJOR GREATER_EQUAL 12)
-        set(CMAKE_CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES} 70 80 90)
+        set(CMAKE_CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES} 90)
     endif()
 endif()
 
@@ -100,50 +100,21 @@ add_custom_target(ut)
 # Details
 add_subdirectory(ark)
 
-# Add header files to library targets
-file(GLOB_RECURSE ARK_HEADERS CONFIGURE_DEPENDS
-    ${CMAKE_SOURCE_DIR}/ark/include/*.h)
-file(GLOB_RECURSE CUTLASS_HEADERS CONFIGURE_DEPENDS
-    ${CMAKE_SOURCE_DIR}/third_party/cutlass/include/*.h
-    ${CMAKE_SOURCE_DIR}/third_party/cutlass/include/*.hpp
-    ${CMAKE_SOURCE_DIR}/third_party/cutlass/include/*.inl)
-target_sources(ark PUBLIC
-    FILE_SET install_headers
-    TYPE HEADERS
-    BASE_DIRS ${CMAKE_SOURCE_DIR}/ark/include
-    FILES ${ARK_HEADERS}
-)
-target_sources(ark_static PUBLIC
-    FILE_SET install_headers
-    TYPE HEADERS
-    BASE_DIRS ${CMAKE_SOURCE_DIR}/ark/include
-    FILES ${ARK_HEADERS}
-)
-target_sources(ark PUBLIC
-    FILE_SET install_cutlass_headers
-    TYPE HEADERS
-    BASE_DIRS ${CMAKE_SOURCE_DIR}/third_party/cutlass/include
-    FILES ${CUTLASS_HEADERS}
-)
-target_sources(ark_static PUBLIC
-    FILE_SET install_cutlass_headers
-    TYPE HEADERS
-    BASE_DIRS ${CMAKE_SOURCE_DIR}/third_party/cutlass/include
-    FILES ${CUTLASS_HEADERS}
-)
-
-# Install libraries
-install(TARGETS ark ark_static
-    LIBRARY DESTINATION ark/lib
-    ARCHIVE DESTINATION ark/lib
-    FILE_SET install_headers DESTINATION ark/include
-    FILE_SET install_cutlass_headers DESTINATION ark/include/kernels
-)
-
-# Install Python module
 if(BUILD_PYTHON)
+    # Install Python module
     add_subdirectory(python)
+    add_dependencies(ark_py build)
+    install(TARGETS ark_py LIBRARY DESTINATION ark)
+else()
+    # Install libraries
+    install(TARGETS ark ark_static
+        LIBRARY DESTINATION ark/lib
+        ARCHIVE DESTINATION ark/lib
+    )
 endif()
 
+# Install header files
+install(DIRECTORY ${BUILD_DIR}/include DESTINATION ark)
+
 # Utils
 include(${PROJECT_SOURCE_DIR}/cmake/Utils.cmake)

ark/include/ark.h

@@ -11,7 +11,7 @@
 
 #define ARK_MAJOR 0
 #define ARK_MINOR 2
-#define ARK_PATCH 0
+#define ARK_PATCH 1
 #define ARK_VERSION (ARK_MAJOR * 10000 + ARK_MINOR * 100 + ARK_PATCH)
 
 namespace ark {

ark/ops/ops_add.cc

@@ -52,7 +52,7 @@ Tensor *Model::add(Tensor *input, Tensor *other, Tensor *output,
 {
     CHECK(input != nullptr);
     CHECK(other != nullptr);
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_div.cc

@@ -53,7 +53,7 @@ Tensor *Model::div(Tensor *input, Tensor *other, Tensor *output,
 {
     assert(input != nullptr);
     assert(other != nullptr);
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_exp.cc

@@ -47,7 +47,7 @@ std::string ExpOp::function_name(const OpConfig &cfg) const
 Tensor *Model::exp(Tensor *input, Tensor *output, const string &name)
 {
     assert(input != nullptr);
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_gelu.cc

@@ -45,7 +45,7 @@ std::string GeluOp::function_name(const OpConfig &cfg) const
 Tensor *Model::gelu(Tensor *input, Tensor *output, const std::string &name)
 {
     assert(input != nullptr);
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_im2col.cc

@@ -85,7 +85,7 @@ Tensor *Model::im2col(Tensor *input, int kernel_height, int kernel_width,
                       Tensor *output, const std::string &name)
 {
     assert(input != nullptr);
-    DimType n, c, h, w;
+    DimType n = 1, c = 1, h = 1, w = 1;
     int input_ndims = input->ndims();
     if (input_ndims == 2) {
         n = 1;
@@ -107,7 +107,7 @@ Tensor *Model::im2col(Tensor *input, int kernel_height, int kernel_width,
             "invalid # of input dimensions. Expected 2, 3, or 4, but given ",
             input_ndims);
     }
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_layernorm.cc

@@ -46,7 +46,7 @@ std::string LayernormOp::function_name(const OpConfig &cfg) const
 Tensor *Model::layernorm(Tensor *input, Tensor *output, const std::string &name)
 {
     assert(input != nullptr);
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (input->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (input->type == FP32) {

ark/ops/ops_matmul.cc

@@ -136,7 +136,7 @@ Tensor *Model::matmul(Tensor *mat_a, Tensor *mat_b, Tensor *mat_y,
         LOG(ERROR, "inner dimensions mismatch: ", k, " and ", k2);
     }
 
-    OpPrecType pt;
+    OpPrecType pt = OP_PREC_NONE;
     if (mat_a->type == FP16) {
         pt = OP_PREC_FP16;
     } else if (mat_a->type == FP32) {