Show device information log for DirectML.

2026-09-20 01:31:13 +02:00 · 2023-10-17 12:24:27 +09:00
parent bb912c35f9
commit 36bef98cd5
2 changed files with 6 additions and 2 deletions
@@ -60,10 +60,9 @@ class DirectML:
        return DeviceProperties(get_device(device))

    def memory_stats(device: Optional[rDevice]=None):
-        mem_stat_fill = "DirectMLDevice"
        return {
            "num_ooms": 0,
-            "num_alloc_retries": mem_stat_fill,
+            "num_alloc_retries": 0,
        }

    mem_get_info: Callable = mem_get_info