diff --git a/openpilot/selfdrive/ui/mici/layouts/home.py b/openpilot/selfdrive/ui/mici/layouts/home.py index 4c4dfac520..d2f22d4a71 100644 --- a/openpilot/selfdrive/ui/mici/layouts/home.py +++ b/openpilot/selfdrive/ui/mici/layouts/home.py @@ -248,8 +248,10 @@ class MiciHomeLayout(Widget): # ***** Center-aligned bottom section icons ***** self._experimental_icon.set_visible(ui_state.experimental_mode) - self._egpu_icon.set_visible(ui_state.usbgpu and ui_state.usbgpu_compiled) - self._egpu_icon_gray.set_visible(ui_state.usbgpu and not ui_state.usbgpu_compiled) + wgpu_running = ui_state.wgpu_enabled and ui_state.sm.alive["modelV2"] and ui_state.sm.valid["modelV2"] + self._egpu_icon.set_visible((ui_state.usbgpu and ui_state.usbgpu_compiled) or wgpu_running) + self._egpu_icon_gray.set_visible((ui_state.usbgpu and not ui_state.usbgpu_compiled) or + (ui_state.wgpu_enabled and not wgpu_running)) self._mic_icon.set_visible(ui_state.recording_audio) self._body_icon.set_visible(bool(ui_state.is_body)) diff --git a/openpilot/selfdrive/ui/ui_state.py b/openpilot/selfdrive/ui/ui_state.py index 59742edfed..d2790a4df1 100644 --- a/openpilot/selfdrive/ui/ui_state.py +++ b/openpilot/selfdrive/ui/ui_state.py @@ -82,6 +82,7 @@ class UIState(UIStateSP): self.experimental_mode: bool = self.params.get_bool("ExperimentalMode") self.usbgpu: bool = self.params.get_bool("UsbGpuPresent") self.usbgpu_compiled: bool = self.params.get_bool("UsbGpuCompiled") + self.wgpu_enabled: bool = self.params.get_bool("WgpuEnabled") self.started: bool = False self.ignition: bool = False self.recording_audio: bool = False @@ -213,6 +214,7 @@ class UIState(UIStateSP): self.experimental_mode = self.params.get_bool("ExperimentalMode") self.usbgpu = self.params.get_bool("UsbGpuPresent") self.usbgpu_compiled = self.params.get_bool("UsbGpuCompiled") + self.wgpu_enabled = self.params.get_bool("WgpuEnabled") UIStateSP.update_params(self)