Expose the version string in C++ and Python (rapidsai#666)

Closes rapidsai#663

Depends on rapidsai#665

* Expose C++ macros `RMM_VER_MAJOR` and `RMM_VER_MINOR` to encode the major and minor versions of RMM. This is useful for downstream packages that links with RMM in the C++ layer.
* Expose the version string via `rmm.__version__`, to follow a widely-recognized convention for accessing version of Python modules.

Authors:
  - Philip Hyunsu Cho (@hcho3)

Approvers:
  - Ashwin Srinath (@shwina)
  - @jakirkham
  - Keith Kraus (@kkraus14)
  - Mark Harris (@harrism)

URL: rapidsai#666

Author: hcho3

CMakeLists.txt

@@ -1,5 +1,5 @@
 # =============================================================================
-# Copyright (c) 2018, NVIDIA CORPORATION.
+# Copyright (c) 2018-2021, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 # in compliance with the License. You may obtain a copy of the License at
@@ -20,6 +20,10 @@ project(
 
 include(cmake/Modules/CPM.cmake)
 include(cmake/Modules/RMM_thirdparty.cmake)
+include(cmake/Modules/Version.cmake)
+
+# Write the version header
+write_version()
 
 # build type
 
@@ -53,8 +57,10 @@ find_package(CUDAToolkit REQUIRED)
 add_library(rmm INTERFACE)
 add_library(rmm::rmm ALIAS rmm)
 
-target_include_directories(rmm INTERFACE "$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>"
-                                         "$<INSTALL_INTERFACE:include>")
+target_include_directories(
+  rmm
+  INTERFACE "$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>"
+            "$<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/include>" "$<INSTALL_INTERFACE:include>")
 
 if(CUDA_STATIC_RUNTIME)
   message(STATUS "RMM: Enabling static linking of cudart")
@@ -98,6 +104,8 @@ set(INSTALL_CONFIGDIR ${CMAKE_INSTALL_LIBDIR}/cmake/rmm)
 install(TARGETS rmm EXPORT rmm-targets)
 
 install(DIRECTORY include/rmm DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
+install(FILES ${RMM_BINARY_DIR}/include/rmm/version_config.hpp
+        DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/rmm)
 
 include(CMakePackageConfigHelpers)
 configure_package_config_file(cmake/rmm-config.cmake.in ${RMM_BINARY_DIR}/rmm-config.cmake

cmake/Modules/Version.cmake

@@ -0,0 +1,18 @@
+# Copyright (c) 2021, NVIDIA CORPORATION.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+# in compliance with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software distributed under the License
+# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+# or implied. See the License for the specific language governing permissions and limitations under
+# the License.
+
+# Generate version_config.hpp from the version found in CMakeLists.txt
+function(write_version)
+  message(STATUS "RMM VERSION: ${RMM_VERSION}")
+  configure_file(${RMM_SOURCE_DIR}/cmake/version_config.hpp.in
+                 ${RMM_BINARY_DIR}/include/rmm/version_config.hpp @ONLY)
+endfunction(write_version)

cmake/version_config.hpp.in

@@ -0,0 +1,20 @@
+/*
+ * Copyright (c) 2021, NVIDIA CORPORATION.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#pragma once
+
+#define RMM_VERSION_MAJOR @RMM_VERSION_MAJOR@
+#define RMM_VERSION_MINOR @RMM_VERSION_MINOR@
+#define RMM_VERSION_PATCH @RMM_VERSION_PATCH@

python/rmm/__init__.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2018-2019, NVIDIA CORPORATION.
+# Copyright (c) 2018-2021, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -24,3 +24,7 @@
     reinitialize,
     rmm_cupy_allocator,
 )
+
+from rmm._version import get_versions
+
+__version__ = get_versions()["version"]