tests: Update for new VFIO worker

Adjust the VFIO device path and the disk image based on the new VFIO CI
worker.

Signed-off-by: Rob Bradford <rbradford@rivosinc.com>
This commit is contained in:
Rob Bradford 2024-10-12 17:59:23 +01:00
parent 79ccb25f78
commit 81db2f0233
2 changed files with 13 additions and 8 deletions

View file

@ -1128,7 +1128,10 @@ impl Guest {
#[cfg(target_arch = "x86_64")]
pub fn check_nvidia_gpu(&self) {
assert!(self.ssh_command("nvidia-smi").unwrap().contains("Tesla T4"));
assert!(self
.ssh_command("nvidia-smi")
.unwrap()
.contains("NVIDIA L40S"));
}
pub fn reboot_linux(&self, current_reboot_count: u32, custom_timeout: Option<i32>) {