| // Copyright 2019 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/metal_util/device_removal.h" |
| |
| #include "base/process/process.h" |
| |
| #import <Metal/Metal.h> |
| |
| namespace metal { |
| |
| void RegisterGracefulExitOnDeviceRemoval() { |
| if (@available(macOS 10.13, *)) { |
| id<NSObject> deviceObserver = nil; |
| MTLCopyAllDevicesWithObserver( |
| &deviceObserver, |
| ^(id<MTLDevice> device, MTLDeviceNotificationName name) { |
| if (name == MTLDeviceRemovalRequestedNotification || |
| name == MTLDeviceWasRemovedNotification) { |
| // Exit the GPU process without error. The browser process sees |
| // this error code as a graceful shutdown, so relaunches the GPU |
| // process without incrementing the crash count. |
| // |
| // Note this wouldn't work nicely with in-process-gpu (it would |
| // exit the browser), but we don't support that on macOS anyway. |
| base::Process::TerminateCurrentProcessImmediately(0); |
| } |
| }); |
| } |
| } |
| |
| } // namespace metal |