tests: nvidia: cc: USE is_confidential_gpu_hw

This function has recently been introduced, so we align patterns.

Signed-off-by: Manuel Huber <manuelh@nvidia.com>
This commit is contained in:
Manuel Huber
2025-12-05 23:20:04 +00:00
committed by Fabiano Fidêncio
parent cd1f55b41c
commit 6e31cf2156
2 changed files with 9 additions and 6 deletions

View File

@@ -11,10 +11,13 @@ load "${BATS_TEST_DIRNAME}/confidential_common.sh"
RUNTIME_CLASS_NAME=${RUNTIME_CLASS_NAME:-kata-qemu-nvidia-gpu}
export RUNTIME_CLASS_NAME
# TODO: Replace with is_confidential_gpu_hardware() once available
KATA_HYPERVISOR=${KATA_HYPERVISOR:-${RUNTIME_CLASS_NAME#kata-}}
export KATA_HYPERVISOR
TEE=false
[[ "${RUNTIME_CLASS_NAME}" = "kata-qemu-nvidia-gpu-snp" ]] && TEE=true
[[ "${RUNTIME_CLASS_NAME}" = "kata-qemu-nvidia-gpu-tdx" ]] && TEE=true
if is_confidential_gpu_hardware; then
TEE=true
fi
export TEE
export POD_NAME_CUDA="nvidia-cuda-vectoradd"

View File

@@ -15,10 +15,10 @@ export LOCAL_NIM_CACHE="/opt/nim/.cache"
SKIP_MULTI_GPU_TESTS=${SKIP_MULTI_GPU_TESTS:-false}
# TODO: Replace with is_confidential_gpu_hardware() once available
TEE=false
[[ "${RUNTIME_CLASS_NAME}" = "kata-qemu-nvidia-gpu-snp" ]] && TEE=true
[[ "${RUNTIME_CLASS_NAME}" = "kata-qemu-nvidia-gpu-tdx" ]] && TEE=true
if is_confidential_gpu_hardware; then
TEE=true
fi
export TEE
POD_NAME_EMBEDQA="nvidia-nim-llama-3-2-nv-embedqa-1b-v2"