Coretex
gpu_memory_usage.py
1
# Copyright (C) 2023 Coretex LLC
2
3
# This file is part of Coretex.ai
4
5
# This program is free software: you can redistribute it and/or modify
6
# it under the terms of the GNU Affero General Public License as
7
# published by the Free Software Foundation, either version 3 of the
8
# License, or (at your option) any later version.
9
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
# GNU Affero General Public License for more details.
14
15
# You should have received a copy of the GNU Affero General Public License
16
# along with this program. If not, see <https://www.gnu.org/licenses/>.
17
18
from
..metric
import
Metric
19
from
.....statistics
import
getGpuMemoryUsage
20
21
22
class
MetricGPUMemoryUsage(Metric):
23
24
def
extract(self) -> float:
25
return
getGpuMemoryUsage()
coretex
entities
task_run
metrics
predefined_metrics
gpu_memory_usage.py
Generated by
1.9.1