Skip to content

Run NVTabular Single GPU tests for CUDA 11 and 12 on RAPIDS Runner #271

Run NVTabular Single GPU tests for CUDA 11 and 12 on RAPIDS Runner

Run NVTabular Single GPU tests for CUDA 11 and 12 on RAPIDS Runner #271

Workflow file for this run

name: GPU Tests
on:
workflow_dispatch:
push:
branches: [main]
tags:
- "v[0-9]+.[0-9]+.[0-9]+"
pull_request:
branches: [main]
types: [opened, synchronize, reopened, closed]
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true
jobs:
gpu-tests:
runs-on: linux-amd64-gpu-v100-latest-1
container:
image: nvidia/cuda:11.8.0-base-ubuntu22.04
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
steps:
- uses: actions/checkout@v3
with:
fetch-depth: 0
- name: Get Branch name
id: get-branch-name
uses: NVIDIA-Merlin/.github/actions/branch-name@main
- name: Run tests
run: |
merlin_branch="${{ steps.get-branch-name.outputs.branch }}"
RAPIDS_VERSION=23.04 MERLIN_BRANCH="$merlin_branch" tox -e test-gpu