Skip to content

Add DRA ResourceClaim tracking for device allocation visualization#519

Open
yahavb wants to merge 1 commit into
awslabs:mainfrom
yahavb:dra-support
Open

Add DRA ResourceClaim tracking for device allocation visualization#519
yahavb wants to merge 1 commit into
awslabs:mainfrom
yahavb:dra-support

Conversation

@yahavb

@yahavb yahavb commented Jun 4, 2026

Copy link
Copy Markdown

Summary

  • Watch ResourceClaim objects to track DRA (Dynamic Resource Allocation) device usage per node
  • Enables visualization of devices managed by DRA drivers (e.g., AWS Neuron) that don't appear in traditional pod resource requests
  • Maps the neuron.aws.com DRA driver to the aws.amazon.com/neuron allocatable resource so the progress bar shows correct utilization
  • Gracefully degrades when the DRA API is not available on the cluster

Usage

# Neuron devices only
eks-node-viewer --resources aws.amazon.com/neuron

# CPU and Neuron devices together
eks-node-viewer --resources cpu,aws.amazon.com/neuron

# With node selector
eks-node-viewer --resources cpu,aws.amazon.com/neuron --node-selector node-type=trn2

# Alias for convenience
alias eks-node-viewer-neuron="eks-node-viewer --resources cpu,aws.amazon.com/neuron"

Example Output

Screenshot-eks-node-viewer-neuron

Test plan

  • Build succeeds with go build ./cmd/eks-node-viewer
  • Existing tests pass with make test
  • Neuron DRA allocation shows correct usage on trn2 nodes
  • Gracefully handles clusters without DRA API enabled
  • Works alongside traditional resource monitoring (cpu, memory)

Watch ResourceClaim objects to track DRA (Dynamic Resource Allocation)
device usage per node. This enables visualization of devices managed by
DRA drivers (e.g., AWS Neuron) that don't appear in traditional pod
resource requests.

Maps the neuron.aws.com DRA driver to the aws.amazon.com/neuron
allocatable resource so the progress bar shows correct utilization.

Usage: eks-node-viewer --resources cpu,aws.amazon.com/neuron
@yahavb yahavb requested a review from a team as a code owner June 4, 2026 16:40
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.

1 participant