Skip to content

Commit

Permalink
Fixing file names to match tags
Browse files Browse the repository at this point in the history
  • Loading branch information
nv-braf committed Nov 26, 2024
1 parent 6f4a126 commit 79389c0
Show file tree
Hide file tree
Showing 14 changed files with 45 additions and 12 deletions.
33 changes: 33 additions & 0 deletions genai-perf/genai_perf/record/types/energy_consumption_avg.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from functools import total_ordering

from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
class GpuEnergyConsumptionAvg(GPUEnergyConsumptionBase):
"""
A record for avg GPU Energy Consumption metric
"""

tag = GPUEnergyConsumptionBase.base_tag + "_avg"

def __init__(self, value, device_uuid=None, timestamp=0):
super().__init__(value, device_uuid, timestamp)

@classmethod
def header(cls, aggregation_tag=False) -> str:
return "Avg. GPU Energy Consumption (MJ)"
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from functools import total_ordering

from genai_perf.record.types.gpu_energy_consumption_base import GPUEnergyConsumptionBase
from genai_perf.record.types.energy_consumption_base import GPUEnergyConsumptionBase


@total_ordering
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,11 @@

from functools import total_ordering

from genai_perf.record.types.gpu_total_memory_base import GPUTotalMemoryBase
from genai_perf.record.types.total_gpu_memory_base import GPUTotalMemoryBase


@total_ordering
class GpuTotalMemoryAvg(GPUTotalMemoryBase):
class GPUTotalMemoryAvg(GPUTotalMemoryBase):
"""
A record for avg GPU Total Memory metric
"""
Expand Down

0 comments on commit 79389c0

Please sign in to comment.