Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
M
MojitOS
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
sepia-pub
MojitOS
Commits
5b6f0ec8
Commit
5b6f0ec8
authored
2 years ago
by
floreal.risso
Browse files
Options
Downloads
Patches
Plain Diff
pedantic
parent
54bba401
No related branches found
No related tags found
2 merge requests
!9
fix sensor example (doc)
,
!5
Add dev name to labels
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
src/nvidia_gpu.c
+30
-25
30 additions, 25 deletions
src/nvidia_gpu.c
src/nvidia_gpu.h
+9
-9
9 additions, 9 deletions
src/nvidia_gpu.h
with
39 additions
and
34 deletions
src/nvidia_gpu.c
+
30
−
25
View file @
5b6f0ec8
...
...
@@ -20,10 +20,14 @@
#include
<stdio.h>
#include
<stdint.h>
#include
<nvml.h>
#include
<stdlib.h>
#include
<string.h>
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wpedantic"
#include
<nvml.h>
#pragma GCC diagnostic pop
#include
"util.h"
// -----------------------------SENSOR_KIND
...
...
@@ -36,7 +40,7 @@ typedef enum {
}
SENSOR_KIND
;
typedef
struct
Device
Device
;
typedef
struct
Nvidia
Sensor
Nvidia
Sensor
;
typedef
struct
Nvidia
Gpu
Nvidia
Gpu
;
typedef
struct
ISensor
ISensor
;
typedef
struct
Sensor
Sensor
;
...
...
@@ -69,8 +73,8 @@ struct Device {
unsigned
int
count
;
};
// -- Nvidia
Sensor
: represents the devices
struct
Nvidia
Sensor
{
// -- Nvidia
Gpu
: represents the devices
struct
Nvidia
Gpu
{
Device
*
devices
;
unsigned
int
count
;
};
...
...
@@ -217,7 +221,7 @@ unsigned int get_memory_sensor(uint64_t *results, const Device *device, void *no
}
unsigned
int
label_memory_sensor
(
char
**
labels
,
void
*
data
)
unsigned
int
label_memory_sensor
(
char
**
labels
,
void
*
data
)
{
MemoryData
*
memory_data
=
(
MemoryData
*
)
data
;
...
...
@@ -229,7 +233,7 @@ unsigned int label_memory_sensor(char **labels, void* data)
}
void
clean_memory_sensor
(
void
*
data
)
{
free
(
data
);
free
(
data
);
}
// ----------------------UTILIZATION_SENSOR
...
...
@@ -242,7 +246,7 @@ typedef enum {
}
UtilizationKind
;
typedef
struct
{
char
labels
[
COUNT_UTILIZATION
][
UTILIZATION_LABEL_SIZE
];
char
labels
[
COUNT_UTILIZATION
][
UTILIZATION_LABEL_SIZE
];
}
UtilizationData
;
static
const
char
*
utilization_names
[
COUNT_UTILIZATION
]
=
{
"gpu"
,
"memory"
};
...
...
@@ -262,7 +266,7 @@ unsigned int init_utilization_sensor(const Device *device, void **data)
UtilizationData
*
utilization_data
=
(
UtilizationData
*
)
calloc
(
1
,
sizeof
(
UtilizationData
));
for
(
unsigned
int
i
=
0
;
i
<
COUNT_UTILIZATION
;
i
++
)
{
snprintf
(
utilization_data
->
labels
[
i
],
UTILIZATION_LABEL_SIZE
,
label_template
,
device_idx
,
utilization_base_name
,
utilization_names
[
i
]);
snprintf
(
utilization_data
->
labels
[
i
],
UTILIZATION_LABEL_SIZE
,
label_template
,
device_idx
,
utilization_base_name
,
utilization_names
[
i
]);
}
*
data
=
(
void
*
)
utilization_data
;
...
...
@@ -286,7 +290,7 @@ unsigned int get_utilization_sensor(uint64_t *results, const Device *device, voi
return
COUNT_UTILIZATION
;
}
unsigned
int
label_utilization_sensor
(
char
**
labels
,
void
*
data
)
unsigned
int
label_utilization_sensor
(
char
**
labels
,
void
*
data
)
{
UtilizationData
*
utilization_data
=
(
UtilizationData
*
)
data
;
...
...
@@ -388,7 +392,7 @@ void clean_device(Device *device)
// ------------------------NVIDIA_INTERFACE
unsigned
int
init_nvidia_
sensor
(
char
*
none
,
void
**
ptr
)
unsigned
int
init_nvidia_
gpu
(
char
*
none
,
void
**
ptr
)
{
UNUSED
(
none
);
UNUSED
(
ptr
);
...
...
@@ -418,7 +422,7 @@ unsigned int init_nvidia_sensor(char *none, void **ptr)
}
}
Nvidia
Sensor
*
nvidia
=
(
Nvidia
Sensor
*
)
calloc
(
1
,
sizeof
(
Nvidia
Sensor
));
Nvidia
Gpu
*
nvidia
=
(
Nvidia
Gpu
*
)
calloc
(
1
,
sizeof
(
Nvidia
Gpu
));
nvidia
->
devices
=
devices
;
nvidia
->
count
=
device_count
;
...
...
@@ -427,9 +431,9 @@ unsigned int init_nvidia_sensor(char *none, void **ptr)
}
unsigned
int
get_nvidia_
sensor
(
uint64_t
*
results
,
void
*
ptr
)
unsigned
int
get_nvidia_
gpu
(
uint64_t
*
results
,
void
*
ptr
)
{
Nvidia
Sensor
*
nvidia
=
(
Nvidia
Sensor
*
)
ptr
;
Nvidia
Gpu
*
nvidia
=
(
Nvidia
Gpu
*
)
ptr
;
unsigned
count
=
0
;
for
(
unsigned
int
i
=
0
;
i
<
nvidia
->
count
;
i
++
)
{
...
...
@@ -441,8 +445,9 @@ unsigned int get_nvidia_sensor(uint64_t *results, void *ptr)
return
count
;
}
unsigned
int
label_nvidia_sensor
(
char
**
labels
,
void
*
ptr
)
{
NvidiaSensor
*
nvidia
=
(
NvidiaSensor
*
)
ptr
;
unsigned
int
label_nvidia_gpu
(
char
**
labels
,
void
*
ptr
)
{
NvidiaGpu
*
nvidia
=
(
NvidiaGpu
*
)
ptr
;
unsigned
count
=
0
;
for
(
unsigned
int
i
=
0
;
i
<
nvidia
->
count
;
i
++
)
{
...
...
@@ -454,9 +459,9 @@ unsigned int label_nvidia_sensor(char **labels, void *ptr) {
return
count
;
}
void
clean_nvidia_
sensor
(
void
*
ptr
)
void
clean_nvidia_
gpu
(
void
*
ptr
)
{
Nvidia
Sensor
*
nvidia
=
(
Nvidia
Sensor
*
)
ptr
;
Nvidia
Gpu
*
nvidia
=
(
Nvidia
Gpu
*
)
ptr
;
for
(
unsigned
int
i
=
0
;
i
<
nvidia
->
count
;
i
++
)
{
clean_device
(
&
nvidia
->
devices
[
i
]);
...
...
@@ -475,26 +480,26 @@ int main()
void
*
ptr
=
NULL
;
char
*
none
=
NULL
;
unsigned
int
sensor_count
=
init_nvidia_
sensor
(
none
,
&
ptr
);
unsigned
int
sensor_count
=
init_nvidia_
gpu
(
none
,
&
ptr
);
Nvidia
Sensor
*
nvidia
=
(
Nvidia
Sensor
*
)
ptr
;
Nvidia
Gpu
*
nvidia
=
(
Nvidia
Gpu
*
)
ptr
;
printf
(
"%d
\n
"
,
nvidia
->
count
);
printf
(
"%u
\n
"
,
sensor_count
);
uint64_t
results
[
sensor_count
];
char
*
labels
[
sensor_count
];
memset
(
results
,
0
,
sensor_count
*
sizeof
(
uint64_t
));
memset
(
labels
,
0
,
sensor_count
*
sizeof
(
char
**
));
memset
(
labels
,
0
,
sensor_count
*
sizeof
(
char
**
));
unsigned
count_label
=
label_nvidia_
sensor
(
labels
,
ptr
);
unsigned
count_get
=
get_nvidia_
sensor
(
results
,
ptr
);
unsigned
count_label
=
label_nvidia_
gpu
(
labels
,
ptr
);
unsigned
count_get
=
get_nvidia_
gpu
(
results
,
ptr
);
printf
(
"total : %u, get : %u, label : %u
\n
"
,
sensor_count
,
count_get
,
count_label
);
for
(
unsigned
int
i
=
0
;
i
<
sensor_count
;
i
++
)
{
printf
(
"%s "
,
labels
[
i
]);
}
...
...
@@ -505,7 +510,7 @@ int main()
printf
(
"
\n
"
);
printf
(
"sensor_count: %d
\n
"
,
sensor_count
);
clean_nvidia_
sensor
(
ptr
);
clean_nvidia_
gpu
(
ptr
);
}
#endif
This diff is collapsed.
Click to expand it.
src/nvidia_gpu.h
+
9
−
9
View file @
5b6f0ec8
...
...
@@ -18,23 +18,23 @@
*******************************************************/
unsigned
int
init_nvidia_
sensor
(
char
*
,
void
**
);
unsigned
int
get_nvidia_
sensor
(
uint64_t
*
results
,
void
*
);
void
clean_nvidia_
sensor
(
void
*
);
void
label_nvidia_
sensor
(
char
**
labels
,
void
*
);
unsigned
int
init_nvidia_
gpu
(
char
*
,
void
**
);
unsigned
int
get_nvidia_
gpu
(
uint64_t
*
results
,
void
*
);
void
clean_nvidia_
gpu
(
void
*
);
void
label_nvidia_
gpu
(
char
**
labels
,
void
*
);
Sensor
nvidia_gpu
=
{
.
init
=
init_nvidia_
sensor
,
.
get
=
get_nvidia_
sensor
,
.
clean
=
clean_nvidia_
sensor
,
.
label
=
label_nvidia_
sensor
,
.
init
=
init_nvidia_
gpu
,
.
get
=
get_nvidia_
gpu
,
.
clean
=
clean_nvidia_
gpu
,
.
label
=
label_nvidia_
gpu
,
.
nb_opt
=
1
,
};
Optparse
nvidia_gpu_opt
[
1
]
=
{
{
.
longname
=
"nvidia"
,
.
longname
=
"nvidia
-gpu
"
,
.
shortname
=
'n'
,
.
argtype
=
OPTPARSE_NONE
,
.
usage_arg
=
NULL
,
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment