Port MiniWeather over the CUDASTF programming model

cudastf/CMakeLists.txt

@@ -0,0 +1,30 @@
+cmake_minimum_required(VERSION 3.14 FATAL_ERROR)
+project(miniWeather CUDA CXX)
+
+include(cmake/CPM.cmake)
+
+# This will automatically clone CCCL from GitHub and make the exported cmake targets available
+CPMAddPackage(
+    NAME CCCL
+    GITHUB_REPOSITORY nvidia/cccl
+    GIT_TAG main
+    # The following is required to make the `CCCL::cudax` target available:
+    OPTIONS "CCCL_ENABLE_UNSTABLE ON"
+)
+
+# Default to building for the GPU on the current system
+if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+  set(CMAKE_CUDA_ARCHITECTURES native)
+endif()
+
+# If you're building an executable
+add_executable(miniWeather_cudastf miniWeather_cudastf.cu)
+
+target_link_libraries(miniWeather_cudastf PUBLIC cuda)
+
+if (CMAKE_CUDA_COMPILER)
+    target_compile_options(miniWeather_cudastf PUBLIC $<$<COMPILE_LANGUAGE:CUDA>:--expt-relaxed-constexpr>)
+    target_compile_options(miniWeather_cudastf PUBLIC $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>)
+endif()
+
+target_link_libraries(miniWeather_cudastf PRIVATE CCCL::CCCL CCCL::cudax)

cudastf/README.md

@@ -0,0 +1,14 @@
+# CUDASTF port of the miniWeather benchmark
+
+CUDASTF is a task-based programming model implemented as a header-only C++
+library. It is shipped as part of NVIDIA's CCCL project
+<https://github.com/nvidia/cccl>.
+
+After identifying physical fields as `logical data`, and expressing
+computations as tasks which operate on these fields, CUDASTF automatically
+manages data, and infers concurrency opportunities.
+
+This example illustrates how to automatically generate CUDA kernels based on
+CUDASTF's `ctx.parallel_for`. It also demonstrates how these constructs can be
+seamlessly spread over multiple GPUs, and how to leverage CUDA graphs to
+improve performance for small problem sizes.

cudastf/cmake/CPM.cmake

@@ -0,0 +1,24 @@
+# SPDX-License-Identifier: MIT
+#
+# SPDX-FileCopyrightText: Copyright (c) 2019-2023 Lars Melchior and contributors
+
+set(CPM_DOWNLOAD_VERSION 0.40.2)
+set(CPM_HASH_SUM "c8cdc32c03816538ce22781ed72964dc864b2a34a310d3b7104812a5ca2d835d")
+
+if(CPM_SOURCE_CACHE)
+  set(CPM_DOWNLOAD_LOCATION "${CPM_SOURCE_CACHE}/cpm/CPM_${CPM_DOWNLOAD_VERSION}.cmake")
+elseif(DEFINED ENV{CPM_SOURCE_CACHE})
+  set(CPM_DOWNLOAD_LOCATION "$ENV{CPM_SOURCE_CACHE}/cpm/CPM_${CPM_DOWNLOAD_VERSION}.cmake")
+else()
+  set(CPM_DOWNLOAD_LOCATION "${CMAKE_BINARY_DIR}/cmake/CPM_${CPM_DOWNLOAD_VERSION}.cmake")
+endif()
+
+# Expand relative path. This is important if the provided path contains a tilde (~)
+get_filename_component(CPM_DOWNLOAD_LOCATION ${CPM_DOWNLOAD_LOCATION} ABSOLUTE)
+
+file(DOWNLOAD
+     https://github.com/cpm-cmake/CPM.cmake/releases/download/v${CPM_DOWNLOAD_VERSION}/CPM.cmake
+     ${CPM_DOWNLOAD_LOCATION} EXPECTED_HASH SHA256=${CPM_HASH_SUM}
+)
+
+include(${CPM_DOWNLOAD_LOCATION})