Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GPU: added hostgpu inventory payload #34429

Draft
wants to merge 11 commits into
base: valeri.pliskin/wms-gpuentity-more-fields
Choose a base branch
from

Conversation

val06
Copy link
Contributor

@val06 val06 commented Feb 25, 2025

What does this PR do?

Adds new payload to collect gpu devices information from the host.
Currently only nvidia devices are supported.

Motivation

Collect host's system information related to GPUs

Describe how you validated your changes

UTs added

Possible Drawbacks / Trade-offs

Additional Notes

Using Workload Meta Store (WMS) as the source to get the list of nvidia gpu devices.

@val06 val06 added changelog/no-changelog team/ebpf-platform qa/done QA done before merge and regressions are covered by tests labels Feb 25, 2025
@val06 val06 requested review from a team as code owners February 25, 2025 15:53
@val06 val06 requested review from hush-hush and pgimalac February 25, 2025 15:53
@val06 val06 marked this pull request as draft February 25, 2025 15:53
@val06 val06 changed the title added hostgpu inventory payload skeleton GPU: added hostgpu inventory payload Feb 25, 2025
Copy link

cit-pr-commenter bot commented Feb 25, 2025

Go Package Import Differences

Baseline: 565c054
Comparison: 228c220

binaryosarchchange
agentlinuxamd64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
agentlinuxarm64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
agentwindowsamd64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
agentdarwinamd64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
agentdarwinarm64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
iot-agentlinuxamd64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
iot-agentlinuxarm64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl
heroku-agentlinuxamd64
+3, -0
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/def
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/fx
+github.com/DataDog/datadog-agent/comp/metadata/hostgpu/impl

@agent-platform-auto-pr
Copy link
Contributor

agent-platform-auto-pr bot commented Feb 25, 2025

Test changes on VM

Use this command from test-infra-definitions to manually test this PR changes on a VM:

inv aws.create-vm --pipeline-id=56907888 --os-family=ubuntu

Note: This applies to commit 228c220

@agent-platform-auto-pr
Copy link
Contributor

agent-platform-auto-pr bot commented Feb 25, 2025

Uncompressed package size comparison

Comparison with ancestor 565c054734e9cc1f2ca02939feb6d99cf96047f4

Diff per package
package diff status size ancestor threshold
datadog-heroku-agent-amd64-deb 0.03MB ⚠️ 443.31MB 443.28MB 0.50MB
datadog-agent-x86_64-rpm 0.02MB ⚠️ 845.46MB 845.43MB 0.50MB
datadog-agent-x86_64-suse 0.02MB ⚠️ 845.46MB 845.43MB 0.50MB
datadog-agent-amd64-deb 0.02MB ⚠️ 835.66MB 835.64MB 0.50MB
datadog-iot-agent-amd64-deb 0.02MB ⚠️ 62.05MB 62.02MB 0.50MB
datadog-iot-agent-x86_64-rpm 0.02MB ⚠️ 62.12MB 62.09MB 0.50MB
datadog-iot-agent-x86_64-suse 0.02MB ⚠️ 62.11MB 62.09MB 0.50MB
datadog-agent-arm64-deb 0.02MB ⚠️ 826.18MB 826.16MB 0.50MB
datadog-agent-aarch64-rpm 0.02MB ⚠️ 835.95MB 835.93MB 0.50MB
datadog-iot-agent-arm64-deb 0.02MB ⚠️ 59.28MB 59.27MB 0.50MB
datadog-iot-agent-aarch64-rpm 0.02MB ⚠️ 59.35MB 59.33MB 0.50MB
datadog-dogstatsd-amd64-deb 0.00MB 39.42MB 39.42MB 0.50MB
datadog-dogstatsd-x86_64-rpm 0.00MB 39.50MB 39.50MB 0.50MB
datadog-dogstatsd-x86_64-suse 0.00MB 39.50MB 39.50MB 0.50MB
datadog-dogstatsd-arm64-deb 0.00MB 37.96MB 37.96MB 0.50MB

Decision

⚠️ Warning

@agent-platform-auto-pr
Copy link
Contributor

agent-platform-auto-pr bot commented Feb 25, 2025

Static quality checks ✅

Please find below the results from static quality gates

Successful checks

Info

Result Quality gate On disk size On disk size limit On wire size On wire size limit
static_quality_gate_agent_deb_amd64 808.73MiB 841.59MiB 195.28MiB 210.72MiB
static_quality_gate_agent_deb_arm64 799.52MiB 830.66MiB 177.04MiB 192.1MiB
static_quality_gate_agent_rpm_amd64 808.74MiB 841.52MiB 198.14MiB 214.52MiB
static_quality_gate_agent_rpm_arm64 799.57MiB 830.66MiB 178.79MiB 193.64MiB
static_quality_gate_agent_suse_amd64 808.61MiB 841.52MiB 198.14MiB 214.52MiB
static_quality_gate_agent_suse_arm64 799.69MiB 830.66MiB 178.79MiB 194.03MiB
static_quality_gate_dogstatsd_deb_amd64 37.67MiB 49.7MiB 9.78MiB 20.6MiB
static_quality_gate_dogstatsd_deb_arm64 36.27MiB 48.1MiB 8.48MiB 19.1MiB
static_quality_gate_dogstatsd_rpm_amd64 37.67MiB 49.7MiB 9.79MiB 20.6MiB
static_quality_gate_dogstatsd_suse_amd64 37.67MiB 49.7MiB 9.79MiB 20.6MiB
static_quality_gate_iot_agent_deb_amd64 59.25MiB 69.0MiB 14.89MiB 24.8MiB
static_quality_gate_iot_agent_deb_arm64 56.61MiB 66.4MiB 12.84MiB 22.8MiB
static_quality_gate_iot_agent_rpm_amd64 59.25MiB 69.0MiB 14.91MiB 24.8MiB
static_quality_gate_iot_agent_rpm_arm64 56.61MiB 66.4MiB 12.85MiB 22.8MiB
static_quality_gate_iot_agent_suse_amd64 59.24MiB 69.0MiB 14.91MiB 24.8MiB
static_quality_gate_docker_agent_amd64 893.1MiB 926.0MiB 298.79MiB 317.43MiB
static_quality_gate_docker_agent_arm64 907.31MiB 939.07MiB 284.52MiB 302.43MiB
static_quality_gate_docker_agent_jmx_amd64 1.07GiB 1.1GiB 373.91MiB 392.54MiB
static_quality_gate_docker_agent_jmx_arm64 1.07GiB 1.1GiB 355.61MiB 373.21MiB
static_quality_gate_docker_dogstatsd_amd64 45.81MiB 57.88MiB 17.28MiB 28.29MiB
static_quality_gate_docker_dogstatsd_arm64 44.45MiB 56.27MiB 16.16MiB 27.06MiB
static_quality_gate_docker_cluster_agent_amd64 264.96MiB 274.78MiB 106.35MiB 116.28MiB
static_quality_gate_docker_cluster_agent_arm64 280.92MiB 290.82MiB 101.18MiB 111.12MiB

Copy link

Regression Detector

Regression Detector Results

Metrics dashboard
Target profiles
Run ID: 3b759cc7-217a-4fe9-acc8-8d7c7d30847e

Baseline: 565c054
Comparison: 228c220
Diff

Optimization Goals: ✅ No significant changes detected

Fine details of change detection per experiment

perf experiment goal Δ mean % Δ mean % CI trials links
quality_gate_logs % cpu utilization +0.85 [-2.06, +3.76] 1 Logs
quality_gate_idle memory utilization +0.46 [+0.42, +0.50] 1 Logs bounds checks dashboard
file_to_blackhole_1000ms_latency_linear_load egress throughput +0.30 [-0.16, +0.75] 1 Logs
uds_dogstatsd_to_api_cpu % cpu utilization +0.27 [-0.59, +1.13] 1 Logs
file_to_blackhole_1000ms_latency egress throughput +0.19 [-0.58, +0.95] 1 Logs
file_to_blackhole_500ms_latency egress throughput +0.05 [-0.73, +0.84] 1 Logs
file_to_blackhole_300ms_latency egress throughput +0.03 [-0.60, +0.66] 1 Logs
file_to_blackhole_0ms_latency_http2 egress throughput +0.02 [-0.83, +0.88] 1 Logs
file_to_blackhole_0ms_latency egress throughput +0.01 [-0.77, +0.79] 1 Logs
file_to_blackhole_100ms_latency egress throughput +0.00 [-0.67, +0.67] 1 Logs
file_to_blackhole_0ms_latency_http1 egress throughput -0.00 [-0.83, +0.82] 1 Logs
tcp_dd_logs_filter_exclude ingress throughput -0.01 [-0.03, +0.01] 1 Logs
uds_dogstatsd_to_api ingress throughput -0.01 [-0.31, +0.28] 1 Logs
quality_gate_idle_all_features memory utilization -0.04 [-0.10, +0.02] 1 Logs bounds checks dashboard
file_tree memory utilization -0.46 [-0.53, -0.40] 1 Logs
tcp_syslog_to_blackhole ingress throughput -0.84 [-0.90, -0.78] 1 Logs

Bounds Checks: ✅ Passed

perf experiment bounds_check_name replicates_passed links
file_to_blackhole_0ms_latency lost_bytes 10/10
file_to_blackhole_0ms_latency memory_usage 10/10
file_to_blackhole_0ms_latency_http1 lost_bytes 10/10
file_to_blackhole_0ms_latency_http1 memory_usage 10/10
file_to_blackhole_0ms_latency_http2 lost_bytes 10/10
file_to_blackhole_0ms_latency_http2 memory_usage 10/10
file_to_blackhole_1000ms_latency memory_usage 10/10
file_to_blackhole_1000ms_latency_linear_load memory_usage 10/10
file_to_blackhole_100ms_latency lost_bytes 10/10
file_to_blackhole_100ms_latency memory_usage 10/10
file_to_blackhole_300ms_latency lost_bytes 10/10
file_to_blackhole_300ms_latency memory_usage 10/10
file_to_blackhole_500ms_latency lost_bytes 10/10
file_to_blackhole_500ms_latency memory_usage 10/10
quality_gate_idle intake_connections 10/10 bounds checks dashboard
quality_gate_idle memory_usage 10/10 bounds checks dashboard
quality_gate_idle_all_features intake_connections 10/10 bounds checks dashboard
quality_gate_idle_all_features memory_usage 10/10 bounds checks dashboard
quality_gate_logs intake_connections 10/10
quality_gate_logs lost_bytes 10/10
quality_gate_logs memory_usage 10/10

Explanation

Confidence level: 90.00%
Effect size tolerance: |Δ mean %| ≥ 5.00%

Performance changes are noted in the perf column of each table:

  • ✅ = significantly better comparison variant performance
  • ❌ = significantly worse comparison variant performance
  • ➖ = no significant change in performance

A regression test is an A/B test of target performance in a repeatable rig, where "performance" is measured as "comparison variant minus baseline variant" for an optimization goal (e.g., ingress throughput). Due to intrinsic variability in measuring that goal, we can only estimate its mean value for each experiment; we report uncertainty in that value as a 90.00% confidence interval denoted "Δ mean % CI".

For each experiment, we decide whether a change in performance is a "regression" -- a change worth investigating further -- if all of the following criteria are true:

  1. Its estimated |Δ mean %| ≥ 5.00%, indicating the change is big enough to merit a closer look.

  2. Its 90.00% confidence interval "Δ mean % CI" does not contain zero, indicating that if our statistical model is accurate, there is at least a 90.00% chance there is a difference in performance between baseline and comparison variants.

  3. Its configuration does not mark it "erratic".

CI Pass/Fail Decision

Passed. All Quality Gates passed.

  • quality_gate_idle, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_idle, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check lost_bytes: 10/10 replicas passed. Gate passed.
  • quality_gate_idle_all_features, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_idle_all_features, bounds check intake_connections: 10/10 replicas passed. Gate passed.

Copy link
Member

@hush-hush hush-hush left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added a few comments, but looks great overall 👍

@@ -325,6 +325,12 @@ Package haagent implements a component to generate the 'ha_agent_metadata' metad

Package host implements a component to generate the 'host' metadata payload (also known as "v5").

### [comp/metadata/hostgpu](https://pkg.go.dev/github.com/DataDog/datadog-agent/comp/metadata/hostgpu)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will there be a non host GPU at some point ? Or could we rename this compmetadata/gpu ?

@@ -0,0 +1,78 @@
# Inventory Host Payload

This package populates some of the Agent-related fields in the `inventories` product in DataDog. More specifically the
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
This package populates some of the Agent-related fields in the `inventories` product in DataDog. More specifically the
This package populates some of the Agent-related fields in the `Resource Catalog` product in DataDog. More specifically the

# Inventory Host Payload

This package populates some of the Agent-related fields in the `inventories` product in DataDog. More specifically the
`host_gpu_agent` table.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
`host_gpu_agent` table.
`host_gpu` table.

Comment on lines +11 to +14
# Content

The `Set` method from the component allows the rest of the codebase to add any information to the payload.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

There is no Set method for this comp:

Suggested change
# Content
The `Set` method from the component allows the rest of the codebase to add any information to the payload.

@@ -0,0 +1,13 @@
// Unless explicitly stated otherwise all files in this repository are licensed
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do we need a mock for this comp ? It doesn't seems to be used by anything directly.

If we need it, please move the mock to comp/metadata/hostgpu/mock (ie: this section).

// Component is the component type.
type Component interface {
// Refresh trigger a new payload to be sent while still respecting the minimal interval between two updates.
Refresh()
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What is going to use this ?

Comment on lines +87 to +88
// Dependencies defines the dependencies for the hostgpu component
type Dependencies struct {
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Cf the doc please rename this Requires.

}

func (gh *gpuHost) getPayload() marshaler.JSONMarshaler {
gh.fillData()
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will this ever changed within the life-cycle of the Agent ? If not we could compute it only once (maybe at initialization ?).

Comment on lines +86 to +94
p := NewGPUHostProvider(
fxutil.Test[Dependencies](
t,
fx.Provide(func() log.Component { return logmock.New(t) }),
fx.Provide(func() workloadmeta.Component { return &wmsMock{} }),
config.MockModule(),
fx.Provide(func() serializer.MetricSerializer { return serializermock.NewMetricSerializer(t) }),
),
)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We want impl folders to not require FX, this allows non-FX project like OTEL to use those implementation (more on this here)

Suggested change
p := NewGPUHostProvider(
fxutil.Test[Dependencies](
t,
fx.Provide(func() log.Component { return logmock.New(t) }),
fx.Provide(func() workloadmeta.Component { return &wmsMock{} }),
config.MockModule(),
fx.Provide(func() serializer.MetricSerializer { return serializermock.NewMetricSerializer(t) }),
),
)
p := NewGPUHostProvider(Requires{
WMeta: &wmsMock{},
Log: logmock.New(t),
Config: configmock.New(t),
Serializer: serializermock.NewMetricSerializer(t),
})

@github-actions github-actions bot added the medium review PR review might take time label Feb 26, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
changelog/no-changelog medium review PR review might take time qa/done QA done before merge and regressions are covered by tests team/ebpf-platform
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants