Skip to content

Harden vGPU release ordering in the instance lifecycle - #322

Open
yummybomb wants to merge 14 commits into
hypeship/vgpu-framework-abstractionfrom
hypeship/generalize-vgpu-device
Open

Harden vGPU release ordering in the instance lifecycle#322
yummybomb wants to merge 14 commits into
hypeship/vgpu-framework-abstractionfrom
hypeship/generalize-vgpu-device

Conversation

@yummybomb

@yummybomb yummybomb commented Jul 28, 2026

Copy link
Copy Markdown
Contributor

Summary

Layer 2 of the vendor VFIO vGPU stack (#366this#363#364#321). Where #366 is a behavior-preserving refactor, this layer changes lifecycle semantics — release ordering and retention — still on mdev-era machinery, before the vendor VFIO backend lands above:

  • Delete releases the vGPU before other teardown — a successful release persists the cleared assignment before later teardown. A failed release keeps the pre-refactor contract: it is logged and the delete continues, since the release guards never destroy a device they cannot prove is unowned and a skipped release is recovered by startup reconciliation. Stopped instances retry a previously failed release.
  • Stop keeps its no-op contract — a failed retained-release on stop logs and retains the assignment metadata instead of failing the stop; assignments are passed to DestroyVGPU as a struct.
  • Forks don't inherit retained assignments — a retained (failed-release) vGPU assignment is stripped from fork metadata so a fork can't release its parent's device.
  • Start persists a stale release immediately — releasing a stale assignment before start writes the cleared metadata at once, so a later start failure can't leave metadata pointing at a device that's already gone.

Testing

  • go build ./..., go vet clean
  • go test -race targeted lib/instances lifecycle suites pass

Note

Medium Risk
Changes ordering and persistence of GPU teardown and restart-policy blocking during delete; incorrect behavior could leak vGPU slots or briefly restart instances mid-delete, though failures remain logged with reconciliation paths.

Overview
This PR tightens vGPU assignment handling across stop/start/delete, fork, and snapshot restore so metadata matches live host state and forks cannot touch a parent's device.

Delete now blocks restart policy before teardown (and reloads metadata so later saves do not undo it), and releases vGPU immediately after the VMM is dead—before network, passthrough devices, or volumes. A successful release is persisted before the rest of cleanup; failures stay best-effort and delete continues.

Stop no longer clears vGPU fields when DestroyVGPU fails; the assignment stays on disk for retry. Stop on an already-stopped instance and start both retry or flush stale releases (start persists the clear before allocating a new vGPU).

Forks (instance and snapshot) call clearStoredVGPUDevice so they keep GPUProfile but never inherit a retained path/UUID. Restore snapshot copies the instance's current vGPU fields from live metadata instead of resurrecting the snapshot payload.

Tests cover delete ordering, fork/snapshot clearing, restore stale vs retained assignments, and stop/start retry behavior.

Reviewed by Cursor Bugbot for commit 2ac1a4b. Bugbot is set up for automated code reviews on this repo. Configure here.

@yummybomb

Copy link
Copy Markdown
Contributor Author

Trimmed over-engineering after an architectural review against the parent task (host kernel 6.8 / Ubuntu 24.04 support):

  • 1c1f52e — dropped the HasMdev/IsAllocated() compatibility shim. VirtualFunction is internal and never serialized in an API response or persisted, so a single Allocated field is enough.

Kept as-is:

  • the delete/stop behavior change (fail and retain assignment metadata when release fails) — orphan reconciliation only runs at startup, so failing the delete is the correct fail-closed behavior;
  • the darwin DestroyVGPU unknown-framework error — it keeps the fail-closed rollback semantics identical on macOS, which the lib/instances tests exercise in CI.

Tests: go test -race ./lib/devices, ./lib/resources, ./lib/hypervisor/qemu, ./lib/hypervisor/cloudhypervisor, and the vGPU-targeted ./lib/instances tests all pass; the remaining lib/instances E2E failures require KVM/images and fail identically on an unmodified checkout.

@yummybomb
yummybomb marked this pull request as ready for review August 5, 2026 19:47
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 8abb859 to d489510 Compare August 5, 2026 19:47
Comment thread lib/instances/stop.go Outdated
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from d489510 to 186cbf3 Compare August 5, 2026 20:24
Comment thread lib/instances/start.go
@yummybomb
yummybomb changed the base branch from main to hypeship/vgpu-framework-abstraction August 6, 2026 19:16
@yummybomb yummybomb changed the title Generalize vGPU device lifecycle Harden vGPU release ordering in the instance lifecycle Aug 6, 2026
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from e458102 to 7e121c6 Compare August 6, 2026 19:26
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch 2 times, most recently from 7e10cf7 to cf815e2 Compare August 7, 2026 14:02
Comment thread lib/instances/snapshot.go
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 8cf6df0 to ffbcfbb Compare August 7, 2026 20:52
Comment thread lib/instances/delete.go
Comment thread lib/instances/delete.go
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 3ec8640 to e43b46f Compare August 10, 2026 18:58
Comment thread lib/instances/snapshot.go
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from e43b46f to a4e00be Compare August 10, 2026 19:08
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch 2 times, most recently from 74862bf to e22b845 Compare August 17, 2026 15:00
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch 2 times, most recently from 82e452f to 7681ba1 Compare August 21, 2026 20:44
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 7681ba1 to c1048a4 Compare August 21, 2026 20:47
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch 2 times, most recently from d05572f to 4ec35a2 Compare August 24, 2026 19:34
…led release

Start now releases any stored assignment before acquiring a new one and
fails the start if that release fails. Stop and delete retain assignment
metadata when release fails instead of clearing it, so a failed release
can be retried later instead of leaking the device.
Fork cloned the source's StoredMetadata wholesale, so an assignment
retained by a failed release during stop was shared with the fork and
either instance's later release could invalidate the other's. Clear the
assignment fields on the fork while keeping GPUProfile; the fork
acquires its own vGPU on start.
Start released a retained assignment but only saved metadata on the
success path, so a failure later in start left on-disk metadata pointing
at a device that was already released. Save right after the release,
matching the retained-release retry on stop.
Restoring a snapshot rehydrated the vGPU assignment fields embedded in
the snapshot metadata. A snapshot taken while an assignment was retained
after a failed release could resurrect that claim after the release
later succeeded, pointing the instance at a device that is gone or
reused. Keep the instance's current assignment instead: device
assignments are host state, not snapshot payload, and a claim retained
at restore time must survive for the next release retry.
Delete goes back to the pre-refactor behavior: a failed vGPU release is
logged and teardown continues, instead of failing the delete and
retaining the instance. The release-side guards still never destroy a
device they cannot prove is unowned, so continuing only tolerates a
leaked slot until startup reconciliation recovers it - the same
tradeoff the mdev path always made. The restart-policy block before
teardown stays: a delete can still fail earlier when the hypervisor
cannot be confirmed dead.
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 4ec35a2 to 2ac1a4b Compare August 24, 2026 20:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants