ClearML/go-nvlib
1
0
Fork 0
mirror of https://github.com/clearml/go-nvlib synced 2025-01-30 18:36:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
198 Commits 2 Branches 0 Tags 1.6 MiB
main
Commit Graph

198 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Kevin Klues
087de4f458 Pass device.GetBrand() through from NVMl and wrap it to print a string 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-26 21:15:51 +00:00
Kevin Klues
649f196fb7 Merge branch 'fix-bug-in-visit-mig-profiles' into 'main' 
Fix bug in heuristic for which MIG profiles to skip

See merge request nvidia/cloud-native/go-nvlib!35
 2023-03-25 22:11:36 +00:00
Kevin Klues
8c50f9f18f Fix bug in heuristic for which MIG profiles to skip 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-25 22:01:20 +00:00
Kevin Klues
500a464b22 Cache mig profiles in devicelib, not just each device 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-25 18:48:18 +00:00
Kevin Klues
13b333d4a6 Merge branch 'add-arch' into 'main' 
Add ability to query device architeture and cuda compute capability

See merge request nvidia/cloud-native/go-nvlib!34
 2023-03-24 15:08:42 +00:00
Kevin Klues
631bde023f Add ability to query device architeture and cuda compute capability 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-24 14:24:19 +00:00
Kevin Klues
2442aa703d Merge branch 'update-mig-profile-parsing' into 'main' 
Update mig-profile parsing / name generation after go-nvml v12.0 bump

See merge request nvidia/cloud-native/go-nvlib!33
 2023-03-24 11:04:23 +00:00
Kevin Klues
642041d1e0 Update mig-profile parsing / name generation after go-nvml v12.0 bump 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-23 19:29:57 +00:00
Kevin Klues
1f178b880d Update glang version in go module to 1.20 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2023-03-23 10:27:47 +00:00
Carlos Eduardo Arango Gutierrez
62b1c915e1 Merge branch 'blossom-ci' into 'main' 
blossom-ci: enable blossom github action

See merge request nvidia/cloud-native/go-nvlib!32
 2023-03-16 15:32:06 +00:00
Carlos Eduardo Arango Gutierrez
5d6447e5ce
blossom-ci: enable blossom github action 
Enable Blossom CI integration by adding the blossom-ci.yml file.

Signed-off-by: Carlos Eduardo Arango Gutierrez <eduardoa@nvidia.com>
 2023-03-16 16:26:08 +01:00
Christopher Desiniotis
4ea7dac0fa Merge branch 'CNT-4021/bump-go-nvml' into 'main' 
Update vendored go-nvml to 12.0

See merge request nvidia/cloud-native/go-nvlib!31
 2023-03-14 18:17:32 +00:00
Christopher Desiniotis
f5ef22d7bb Update vendored go-nvml to 12.0 
Signed-off-by: Christopher Desiniotis <cdesiniotis@nvidia.com>
 2023-03-14 10:36:13 -07:00
Evan Lezar
95328d8c44 Merge branch 'add-wsl2-detection' into 'main' 
Add HasDXCore to info package

See merge request nvidia/cloud-native/go-nvlib!30
 2023-02-09 14:37:38 +00:00
Evan Lezar
bcbaf5a0de Add HasDXCore to info package 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2023-02-08 16:04:35 +01:00
Kevin Klues
6fe07bb333 Merge branch 'add-more-apis' into 'main' 
Add new APIs in nvml and nvlib.device

See merge request nvidia/cloud-native/go-nvlib!29
 2023-01-19 11:47:11 +00:00
Kevin Klues
264c5dab79 Add NewDeviceByUUID() and NewMigDeviceByUUID() calls to nvlib.device 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-12-08 14:53:50 +00:00
Kevin Klues
5d4be6ac55 Regenerate mocks for NVML 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-12-08 14:53:45 +00:00
Kevin Klues
6a4886e49e Add Placement related calls for GPUInstances in nvml wrapper 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-12-08 14:53:39 +00:00
Kevin Klues
49dcad67c4 Merge branch 'skip-DGX-display' into 'main' 
Skip DGX Display devices in addition to NVIDIA DGX Display devices

See merge request nvidia/cloud-native/go-nvlib!28
 2022-12-07 11:20:48 +00:00
Evan Lezar
7e5501f6a3 Skip DGX Display devices in addition to NVIDIA DGX Display devices 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-12-07 11:40:09 +01:00
Evan Lezar
a27e593595 Merge branch 'skip-display-devices-on-name' into 'main' 
Skip display devices based on model name

See merge request nvidia/cloud-native/go-nvlib!26
 2022-11-21 20:39:40 +00:00
Evan Lezar
417a5254a4 Pin moq version to v0.2.7 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-21 15:50:01 +01:00
Evan Lezar
d69a94ffdd Add .shell target for interactive make 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-21 15:47:46 +01:00
Evan Lezar
1fc1eee392 Remove WithSelecteDeviceClasses option 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-21 15:47:44 +01:00
Evan Lezar
655eb9795c Skip display devices based on device names 
This allows devices to be skipped based on device names and
skips "NVIDIA DGX Display" devices by default.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-21 15:46:15 +01:00
Evan Lezar
0e10f084d1 Merge branch 'fix-pci-id-case' into 'main' 
Ensure pci bus ID is lower case

See merge request nvidia/cloud-native/go-nvlib!25
 2022-11-16 11:13:58 +00:00
Evan Lezar
fa5d0408ce Ensure pci bus ID is lower case 
The PCI Bus ID returned by NVML is upper case and results in the following error:

error getting PCI device class for device:
failed to construct PCI device:
unable to read PCI device vendor id for 0000:0A:00.0:
open /sys/bus/pci/devices/0000:0A:00.0/vendor:
no such file or directory

Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-16 12:12:07 +01:00
Kevin Klues
9110850748 Merge branch 'skip-display-devices' into 'main' 
Skip devices based on PCI device class

See merge request nvidia/cloud-native/go-nvlib!24
 2022-11-16 09:48:48 +00:00
Evan Lezar
4a0fdc2e8a Skip pkg/nvml folder when linting 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-16 10:30:49 +01:00
Evan Lezar
e37e145458 Add filtering of devices based on PCI device class 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-16 10:30:49 +01:00
Evan Lezar
f156c34310 Add private constructor for creating a device 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-15 17:42:22 +01:00
Evan Lezar
e96d9c58f1 Add GetGPUByPciBusID to nvpci.Interface 
This change adds a GetGPUByPciBusID method to the nvpci Interface.
The exising NewDevice function is moved to nvmdev where it is used.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-11-15 17:42:22 +01:00
Zvonko Kaiser
0e8a479bd5 Merge branch 'pciids' into 'main' 
Added PCI IDS support and DPU detection

See merge request nvidia/cloud-native/go-nvlib!23
 2022-11-03 09:57:58 +00:00
Zvonko Kaiser
f3102f8dcb Added PCI IDS support and DPU detection 2022-11-02 03:58:13 -07:00
Evan Lezar
7222fea1a7 Merge branch 'fix-walk-mig-profile' into 'main' 
Ensure that invalid MIG profiles are skipped

See merge request nvidia/cloud-native/go-nvlib!21
 2022-10-14 10:08:48 +00:00
Evan Lezar
8b5e3d224d Ensure that invalid MIG profiles are skipped 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-10-14 10:31:50 +02:00
Kevin Klues
1049a7fa76 Merge branch 'add-events' into 'main' 
Add functions for interacting with Events

See merge request nvidia/cloud-native/go-nvlib!20
 2022-09-22 13:34:27 +00:00
Evan Lezar
1cb5426db8 Add functions for interacting with Events 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-09-21 15:10:06 +02:00
Evan Lezar
01649c65ea Merge branch 'add-device-apis' into 'main' 
Move MIG apis to device package and add extended APIs for top-level devices to it

See merge request nvidia/cloud-native/go-nvlib!19
 2022-09-16 14:15:32 +00:00
Kevin Klues
f933892965 Add extended APIs for top-level devices to the device package 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-09-16 13:34:17 +00:00
Kevin Klues
1d680a93b6 Move MIG apis to device package 
We decided it makes sense to have top level device and MIG device abstractions
all under one package rather than trying to separate them. It will make it
easier to hav them clal between each other without package dependency loops.

Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-09-16 13:09:09 +00:00
Kevin Klues
8719e258a8 Merge branch 'add-mig-pkg' into 'main' 
Add MIG package with abstraction for MIG profiles in it

See merge request nvidia/cloud-native/go-nvlib!15
 2022-09-16 08:17:41 +00:00
Kevin Klues
8e749776c5 Add nvml wrappers for getting GIs and CIs by ID 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-09-15 17:08:00 +00:00
Kevin Klues
e95e3a5e8b Add a MIG package as a subpackage to nvlib 
For now this package only has functions to work with MIG profiles. More
functionality will be added here in the future.

Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-09-15 17:08:00 +00:00
Evan Lezar
16ab19d8ae Merge branch 'add-nvlib-base' into 'main' 
Add a new nvlib package and move the nvinfo package into it

See merge request nvidia/cloud-native/go-nvlib!16
 2022-09-15 11:36:25 +00:00
Kevin Klues
d23f460ad3 Move the nvinfo package into pkg/nvlib/info 
Also build an interface around the API so that it can more easily be mocked.

Signed-off-by: Kevin Klues <kklues@nvidia.com>
 2022-09-15 11:30:34 +00:00
Evan Lezar
a880a67681 Merge branch 'add-get-cuda-driver-version' into 'main' 
Add additional functions to nvml interfaces

See merge request nvidia/cloud-native/go-nvlib!18
 2022-09-05 13:31:28 +00:00
Evan Lezar
211a8eb973 Address minor lint error 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-09-02 15:12:23 +02:00
Evan Lezar
bf9a4d3476 Sort functions in interface alphabetically 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
 2022-09-02 15:07:39 +02:00