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 instance lifecycle semantics for GPU teardown, restart-policy blocking on delete, and snapshot restore metadata—incorrect ordering could leak vGPU slots or double-release, though failures remain best-effort with reconciliation.

Overview
Hardens vGPU assignment handling across delete, stop, start, fork, and snapshot restore so host GPU state and on-disk metadata stay consistent when release fails or clones are created.

Delete now blocks restart policy before teardown (with a metadata reload so the block is not overwritten), and releases vGPU immediately after the VMM is gone—before network, devices, or volumes. A successful release is persisted; failures are logged and teardown still continues.

Stop no longer clears assignment metadata when DestroyVGPU fails; the assignment is retained for retry. Stopping an already-stopped instance retries release via releaseRetainedVGPULocked without breaking the no-op contract.

Start releases any stale retained assignment up front and saves cleared metadata before allocating a new vGPU, so a later start failure cannot leave metadata pointing at an already-destroyed device.

Forks (instance and snapshot) strip device path/UUID from fork metadata while keeping GPUProfile; restore keeps the instance’s current vGPU fields instead of resurrecting values embedded in the snapshot.

Lifecycle tests cover ordering, fork isolation, restore behavior, and best-effort delete on release failure.

Reviewed by Cursor Bugbot for commit 0fccdc8. 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 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
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 4ec35a2 to 2ac1a4b Compare August 24, 2026 20:31
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 2ac1a4b to 33f2738 Compare August 26, 2026 19:37
@yummybomb
yummybomb force-pushed the hypeship/generalize-vgpu-device branch from 33f2738 to b7b5648 Compare August 27, 2026 17:29
…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 b7b5648 to 0fccdc8 Compare August 28, 2026 17:28
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