Merge branch 'main' into 2042-add-internal-wrapper-for-cuda-driver-apis

NVIDIA · Jul 30, 2024 · 3f93e77 · 3f93e77
2 parents 242e135 + c0cfbd0
commit 3f93e77
Show file tree

Hide file tree

Showing 231 changed files with 5,890 additions and 3,848 deletions.
diff --git a/.clang-format b/.clang-format
@@ -117,12 +117,15 @@ IncludeCategories:
   - Regex:           '^<thrust/'
     Priority:            3
     SortPriority:        2
+  - Regex:           '^<cuda/experimental'
+    Priority:            5
+    SortPriority:        4
   - Regex:           '^<cuda/'
     Priority:            4
     SortPriority:        3
   - Regex:           '^<[a-z_]*>$'
-    Priority:            5
-    SortPriority:        4
+    Priority:            6
+    SortPriority:        5
   - Regex:           '^<cuda'
     Priority:            0
     SortPriority:        0

diff --git a/.devcontainer/README.md b/.devcontainer/README.md
@@ -63,6 +63,7 @@ For more information about the sccache configuration and authentication, see the
 ## Quickstart: Docker (Manual Approach) <a name="docker"></a>
 
 ### Prerequisites
+- [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html)
 - [Docker](https://docs.docker.com/desktop/install/linux-install/)
 
 ### Steps

diff --git a/.devcontainer/cuda11.1-gcc6/devcontainer.json b/.devcontainer/cuda11.1-gcc6/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc6-cuda11.1",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc6-cuda11.1",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda11.1-gcc7/devcontainer.json b/.devcontainer/cuda11.1-gcc7/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc7-cuda11.1",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc7-cuda11.1",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda11.1-gcc8/devcontainer.json b/.devcontainer/cuda11.1-gcc8/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc8-cuda11.1",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc8-cuda11.1",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda11.1-gcc9/devcontainer.json b/.devcontainer/cuda11.1-gcc9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc9-cuda11.1",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc9-cuda11.1",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda11.1-llvm9/devcontainer.json b/.devcontainer/cuda11.1-llvm9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm9-cuda11.1",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm9-cuda11.1",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda11.8-gcc11/devcontainer.json b/.devcontainer/cuda11.8-gcc11/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc11-cuda11.8",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc11-cuda11.8",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-gcc10/devcontainer.json b/.devcontainer/cuda12.0-gcc10/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc10-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc10-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-gcc11/devcontainer.json b/.devcontainer/cuda12.0-gcc11/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc11-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc11-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-gcc12/devcontainer.json b/.devcontainer/cuda12.0-gcc12/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc12-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc12-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-gcc9/devcontainer.json b/.devcontainer/cuda12.0-gcc9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc9-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc9-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm10/devcontainer.json b/.devcontainer/cuda12.0-llvm10/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm10-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm10-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm11/devcontainer.json b/.devcontainer/cuda12.0-llvm11/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm11-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm11-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm12/devcontainer.json b/.devcontainer/cuda12.0-llvm12/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm12-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm12-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm13/devcontainer.json b/.devcontainer/cuda12.0-llvm13/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm13-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm13-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm14/devcontainer.json b/.devcontainer/cuda12.0-llvm14/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm14-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm14-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.0-llvm9/devcontainer.json b/.devcontainer/cuda12.0-llvm9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm9-cuda12.0",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm9-cuda12.0",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.2-rapids-conda b/.devcontainer/cuda12.2-rapids-conda
diff --git a/.devcontainer/cuda12.5-gcc10/devcontainer.json b/.devcontainer/cuda12.5-gcc10/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc10-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc10-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc11/devcontainer.json b/.devcontainer/cuda12.5-gcc11/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc11-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc11-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc12/devcontainer.json b/.devcontainer/cuda12.5-gcc12/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc12-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc12-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc13/devcontainer.json b/.devcontainer/cuda12.5-gcc13/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc13-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc13-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc7/devcontainer.json b/.devcontainer/cuda12.5-gcc7/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc7-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc7-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc8/devcontainer.json b/.devcontainer/cuda12.5-gcc8/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc8-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc8-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-gcc9/devcontainer.json b/.devcontainer/cuda12.5-gcc9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc9-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc9-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm10/devcontainer.json b/.devcontainer/cuda12.5-llvm10/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm10-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm10-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm11/devcontainer.json b/.devcontainer/cuda12.5-llvm11/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm11-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm11-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm12/devcontainer.json b/.devcontainer/cuda12.5-llvm12/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm12-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm12-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm13/devcontainer.json b/.devcontainer/cuda12.5-llvm13/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm13-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm13-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm14/devcontainer.json b/.devcontainer/cuda12.5-llvm14/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm14-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm14-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm15/devcontainer.json b/.devcontainer/cuda12.5-llvm15/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm15-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm15-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm16/devcontainer.json b/.devcontainer/cuda12.5-llvm16/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm16-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm16-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm17/devcontainer.json b/.devcontainer/cuda12.5-llvm17/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm17-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm17-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-llvm9/devcontainer.json b/.devcontainer/cuda12.5-llvm9/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-llvm9-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-llvm9-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-oneapi2023.2.0/devcontainer.json b/.devcontainer/cuda12.5-oneapi2023.2.0/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-oneapi2023.2.0-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-oneapi2023.2.0-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.devcontainer/cuda12.5-rapids-conda b/.devcontainer/cuda12.5-rapids-conda
@@ -0,0 +1 @@
+../ci/rapids/cuda12.5-conda
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -1,6 +1,6 @@
 {
   "shutdownAction": "stopContainer",
-  "image": "rapidsai/devcontainers:24.08-cpp-gcc13-cuda12.5",
+  "image": "rapidsai/devcontainers:24.10-cpp-gcc13-cuda12.5",
   "hostRequirements": {
     "gpu": "optional"
   },

diff --git a/.github/actions/workflow-build/action.yml b/.github/actions/workflow-build/action.yml
@@ -25,6 +25,15 @@ inputs:
     description: "Path to the matrix parser script (default if blank: build-workflow.py from action dir)"
     default: ""
     required: false
+  slack_token:
+    description: "The Slack token to use for notifications. No notifications will be sent if not provided."
+    required: false
+  slack_log:
+    description: "Slack channel ID for verbose notifications."
+    required: false
+  slack_alert:
+    description: "Slack channel ID for alert notifications."
+    required: false
 
 outputs:
   workflow:
@@ -35,6 +44,20 @@ runs:
   using: "composite"
   steps:
 
+    - name: Send Slack log notification
+      if: ${{inputs.slack_token != '' && inputs.slack_log != '' }}
+      uses: slackapi/[email protected]
+      env:
+        SLACK_BOT_TOKEN: ${{ inputs.slack_token }}
+        WORKFLOW_TYPE: ${{ github.workflow }} # nightly, weekly, pr, etc.
+        SUMMARY_URL: https://github.com/${{github.repository}}/actions/runs/${{github.run_id}}
+      with:
+        channel-id: ${{ inputs.slack_log }}
+        slack-message: |
+          Workflow '${{ env.WORKFLOW_TYPE }}' starting...
+
+          Details: ${{ env.SUMMARY_URL }}
+
     - name: Inspect changes
       if: ${{ inputs.inspect_changes_script != '' && inputs.inspect_changes_base_sha != '' }}
       id: inspect-changes
@@ -99,3 +122,18 @@ runs:
         name: workflow
         path: workflow/
         compression-level: 0
+
+    - name: Send Slack error notification
+      if: ${{ failure() && inputs.slack_token != '' && (inputs.slack_alert != '' || inputs.slack_log != '') }}
+      uses: slackapi/[email protected]
+      env:
+        SLACK_BOT_TOKEN: ${{ inputs.slack_token }}
+        WORKFLOW_TYPE: ${{ github.workflow }} # nightly, weekly, pr, etc.
+        SUMMARY_URL: https://github.com/${{github.repository}}/actions/runs/${{github.run_id}}
+        CHANNEL_SEP: ${{ (inputs.slack_log != '' && inputs.slack_alert != '') && ',' || ''}}
+      with:
+        channel-id: '${{ inputs.slack_log }}${{env.CHANNEL_SEP}}${{ inputs.slack_alert }}'
+        slack-message: |
+          Workflow '${{ env.WORKFLOW_TYPE }}' encountered an error while preparing to run.
+
+          Details: ${{ env.SUMMARY_URL }}