Evan Lezar
fb0dc9d525
Add ComputeMode constants
...
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2024-03-12 17:23:12 +02:00
Evan Lezar
2feaa48250
Add SetComputeMode method to Device
...
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2024-03-11 16:25:04 +02:00
Evan Lezar
06cbc571ef
Add nvmlDeviceHandle function to Device interface
...
This change allows the underlying device handle to be returned
without relying on type-casting.
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2024-01-09 14:05:01 +01:00
Evan Lezar
80d61efe5d
Add functions related to NVLink info
...
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2023-11-14 12:55:16 +01:00
Evan Lezar
30ca72faaf
TOFIX: Allow libname to be specified
...
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2023-10-23 14:09:03 +02:00
Evan Lezar
278851d719
Use GetLibrary().Lookup() in nvml package
...
This change uses the GetLibrary().Lookup() function in the nvml package
to check whether a particular function is available. This avoids
the need to explicitly open a library, for example.
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2023-10-20 16:56:56 +02:00
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
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
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
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
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
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
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
Evan Lezar
a404873b12
Add additional functions to Device interface
...
Add the following functions to the Device interface:
* GetCudaComputeCapability
* GetAttributes
* GetName
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2022-09-02 15:05:12 +02:00
Evan Lezar
9175bde20b
Add SystemGetCudaDriverVersion to NVML interface
...
Signed-off-by: Evan Lezar <elezar@nvidia.com>
2022-09-01 14:05:16 +02:00
Kevin Klues
2e1e2e784a
Add String() and Error() functions to Return type in nvml package
...
There is a default implementation for these that is overwritten if the
underlying NVML library ends up being used.
Signed-off-by: Kevin Klues <kklues@nvidia.com>
2022-08-11 12:13:41 +00:00
Kevin Klues
008aa70bc6
Add an interface based wrapper around go-nvml for better mocking
...
Signed-off-by: Kevin Klues <kklues@nvidia.com>
2022-08-10 15:52:16 +00:00
