mirror of
https://github.com/NVIDIA/OpenShell.git
synced 2026-10-18 05:39:50 +08:00
Evan Lezar
8c0153438c
* test(e2e): add GPU workload image artifacts Closes #1476 Define local GPU workload image sources for smoke-pass, smoke-fail, and cuda-basic validation, plus a mise build task that tags images with the source revision and records the latest local image refs. Signed-off-by: Evan Lezar <elezar@nvidia.com> * test(e2e): add workload manifest build flow Signed-off-by: Evan Lezar <elezar@nvidia.com> * test(e2e): fix gpu workload README lint Signed-off-by: Evan Lezar <elezar@nvidia.com> * test(e2e): address gpu workload review feedback Signed-off-by: Evan Lezar <elezar@nvidia.com> --------- Signed-off-by: Evan Lezar <elezar@nvidia.com>
52 lines
1.7 KiB
Markdown
52 lines
1.7 KiB
Markdown
<!-- SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. -->
|
|
<!-- SPDX-License-Identifier: Apache-2.0 -->
|
|
|
|
# GPU workload CUDA basic
|
|
|
|
`cuda-basic` validates that a GPU-enabled environment can run a basic CUDA
|
|
runtime workload. It is a single image that runs two validation steps:
|
|
|
|
1. `deviceQuery` checks CUDA runtime, driver, and device discovery.
|
|
2. `vectorAdd` checks kernel launch, device memory allocation, host/device
|
|
copies, synchronization, and result validation.
|
|
|
|
The image builds the samples from `NVIDIA/cuda-samples` tag `v12.8` with a CUDA
|
|
12.8 builder image, then copies only the compiled binaries into the OpenShell
|
|
community base final image.
|
|
|
|
The workload prints `OPENSHELL_GPU_WORKLOAD_SUCCESS` only after both samples
|
|
pass. On failure it prints `OPENSHELL_GPU_WORKLOAD_FAILURE` and exits non-zero.
|
|
|
|
Build it with:
|
|
|
|
```shell
|
|
mise run e2e:workloads:build
|
|
```
|
|
|
|
That command also refreshes the local workload manifest at
|
|
`e2e/gpu/images/.build/workloads.yaml`.
|
|
|
|
To build only this workload locally, set:
|
|
|
|
```shell
|
|
OPENSHELL_GPU_WORKLOAD_IMAGES=cuda-basic mise run e2e:workloads:build
|
|
```
|
|
|
|
Run it directly with Docker CDI:
|
|
|
|
```shell
|
|
source e2e/gpu/images/.build/latest.env
|
|
docker run --rm --device nvidia.com/gpu=all \
|
|
"${OPENSHELL_E2E_GPU_CUDA_WORKLOAD_IMAGE}"
|
|
```
|
|
|
|
Use `podman run` with the same `--device nvidia.com/gpu=all` option when Podman
|
|
CDI is configured.
|
|
|
|
The image does not vendor GPU driver libraries such as `libcuda.so.1`. Those
|
|
libraries must be provided by the host GPU runtime or CDI injection.
|
|
|
|
The CUDA samples are redistributed under the NVIDIA CUDA samples license. The
|
|
license text is copied into the image at
|
|
`/usr/local/share/doc/openshell-gpu-workload/cuda-samples.LICENSE`.
|