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

Upgrade to latest NVIDIA NVML package #4469

Merged
merged 1 commit into from
Jan 14, 2025
Merged

Upgrade to latest NVIDIA NVML package #4469

merged 1 commit into from
Jan 14, 2025

Conversation

harishxr
Copy link
Contributor

@harishxr harishxr commented Jan 8, 2025

Summary

This pull request upgrades the NVIDIA NVML package to the latest version in the nvidia_gpu_manager.go file and significantly refactors the GPU manager unit tests.

Implementation details

  • Updated the NVIDIA NVML package dependency to the latest version (go-nvml v0.x.x).
  • Refactored existing unit tests to use mock objects instead of direct NVML calls.
  • Introduced new mock objects for NVML devices and interfaces using gomock.

Testing

All existing unit tests were updated and verified to pass with the new NVML package version.

New tests cover the changes: yes

Description for the changelog

Upgrade to latest NVIDIA NVML package

Additional Information

Does this PR include breaking model changes? If so, Have you added transformation functions?

Does this PR include the addition of new environment variables in the README?

Licensing

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

@harishxr harishxr marked this pull request as ready for review January 8, 2025 20:39
@harishxr harishxr requested a review from a team as a code owner January 8, 2025 20:39
ecs-init/gpu/nvidia_gpu_manager.go Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager.go Outdated Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager.go Outdated Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager_test.go Outdated Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager_test.go Outdated Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager_test.go Show resolved Hide resolved
singholt
singholt previously approved these changes Jan 14, 2025
ecs-init/gpu/nvidia_gpu_manager.go Outdated Show resolved Hide resolved
ecs-init/gpu/nvidia_gpu_manager_test.go Outdated Show resolved Hide resolved
@harishxr harishxr merged commit 12468ff into aws:dev Jan 14, 2025
40 checks passed
@harishxr
Copy link
Contributor Author

This PR fixes #4466

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.

4 participants