From a291a42abc98d3a38906d8d097aae7a8dc93ec92 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 27 Mar 2024 17:28:06 +1100
Subject: [PATCH] feat: display torch device on startup

This functionality disappeared at some point.
---
 invokeai/app/api_app.py          | 4 ++++
 invokeai/backend/util/devices.py | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/invokeai/app/api_app.py b/invokeai/app/api_app.py
index 39644c4870..333b9a58c0 100644
--- a/invokeai/app/api_app.py
+++ b/invokeai/app/api_app.py
@@ -26,6 +26,7 @@ from invokeai.app.api.no_cache_staticfiles import NoCacheStaticFiles
 from invokeai.app.invocations.model import ModelIdentifierField
 from invokeai.app.services.config.config_default import get_config
 from invokeai.app.services.session_processor.session_processor_common import ProgressImage
+from invokeai.backend.util.devices import get_torch_device_name
 
 from ..backend.util.logging import InvokeAILogger
 from .api.dependencies import ApiDependencies
@@ -60,6 +61,9 @@ logger = InvokeAILogger.get_logger(config=app_config)
 mimetypes.add_type("application/javascript", ".js")
 mimetypes.add_type("text/css", ".css")
 
+torch_device_name = get_torch_device_name()
+logger.info(f"Using torch device: {torch_device_name}")
+
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
diff --git a/invokeai/backend/util/devices.py b/invokeai/backend/util/devices.py
index e02015614e..44d25de025 100644
--- a/invokeai/backend/util/devices.py
+++ b/invokeai/backend/util/devices.py
@@ -27,6 +27,9 @@ def choose_torch_device() -> torch.device:
     else:
         return torch.device(config.device)
 
+def get_torch_device_name() -> str:
+    device = choose_torch_device()
+    return torch.cuda.get_device_name(device) if device.type == "cuda" else device.type.upper()
 
 # We are in transition here from using a single global AppConfig to allowing multiple
 # configurations. It is strongly recommended to pass the app_config to this function.