File tree Expand file tree Collapse file tree 2 files changed +16
-4
lines changed Expand file tree Collapse file tree 2 files changed +16
-4
lines changed Original file line number Diff line number Diff line change @@ -26,11 +26,22 @@ void update_temperature(void)
26
26
void get_serial_number (unsigned int devIdx, char * serial)
27
27
{
28
28
#if (ENABLE_NVML==1)
29
- nvmlDevice_t devHandle;
30
- NVML_CHECK (nvmlDeviceGetHandleByIndex ( devIdx, &devHandle ));
29
+ try
30
+ {
31
+ nvmlDevice_t devHandle;
32
+ NVML_CHECK (nvmlDeviceGetHandleByIndex ( devIdx, &devHandle ));
31
33
32
- unsigned int serialLength = NVML_DEVICE_SERIAL_BUFFER_SIZE;
33
- NVML_CHECK (nvmlDeviceGetSerial ( devHandle, serial, serialLength ));
34
+ unsigned int serialLength = NVML_DEVICE_SERIAL_BUFFER_SIZE;
35
+ NVML_CHECK (nvmlDeviceGetSerial ( devHandle, serial, serialLength ));
36
+ }
37
+ catch (const std::runtime_error& e)
38
+ {
39
+ strncpy (
40
+ serial,
41
+ " unknown (NVML runtime error)" ,
42
+ NVML_DEVICE_SERIAL_BUFFER_SIZE);
43
+ serial[NVML_DEVICE_SERIAL_BUFFER_SIZE-1 ] = ' \0 ' ;
44
+ }
34
45
#else
35
46
(void )(devIdx);
36
47
(void )(serial);
Original file line number Diff line number Diff line change 2
2
3
3
#include <cstdlib>
4
4
#include <cassert>
5
+ #include <cstring>
5
6
#include "cuda_memtest.h"
6
7
7
8
void update_temperature (void );
You can’t perform that action at this time.
0 commit comments