From 36bef98cd56a3b3efaa393bb8d5ebe6d7f54f687 Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Tue, 17 Oct 2023 12:24:27 +0900
Subject: [PATCH] Show device information log for DirectML.

---
 modules/devices.py     | 5 +++++
 modules/dml/backend.py | 3 +--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/modules/devices.py b/modules/devices.py
index 5fe08b73d..1102b9a74 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -45,6 +45,11 @@ def get_gpu_info():
                     'device': get_openvino_device(),
                     'openvino': get_package_version("openvino"),
                 }
+            elif shared.cmd_opts.use_directml:
+                return {
+                    'device': f'{torch.cuda.get_device_name(torch.cuda.current_device())} n={torch.cuda.device_count()}',
+                    'directml': get_package_version("torch-directml"),
+                }
             else:
                 return {}
         except Exception:
diff --git a/modules/dml/backend.py b/modules/dml/backend.py
index 887991e86..16fd5231b 100644
--- a/modules/dml/backend.py
+++ b/modules/dml/backend.py
@@ -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