Register TRACE loglevel ahead of parse_args to avoid error in --specs printout

Signed-off-by: coleramos425 <colramos@amd.com>
2024-03-29 12:19:24 -05:00
parent f7394bf272
commit 706afa7f37
2 changed files with 10 additions and 7 deletions
@@ -353,15 +353,17 @@ class MachineSpecs:
    gpu_l1: str = field(
        default=None,
        metadata={
-            "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs in the system in KiB",
+            "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs.",
            "name": "GPU L1",
+            "unit": "KiB",
        },
    )
    gpu_l2: str = field(
        default=None,
        metadata={
-            "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs in the system in KiB",
+            "doc": "The size of the vL1D cache (per compute-unit) on the accelerators/GPUs.",
            "name": "GPU L2",
+            "unit": "KiB",
        },
    )
    cu_per_gpu: str = field(