123456789101112131415161718192021222324252627282930 |
- // 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() {
- 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
|