Skip to content

Commit

Permalink
[GPU] Add metal to DispatchKeySet (#46455)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #46455

After this PR(#46236 ) landed, the `aten::copy_` can no longer be dispatched to Metal kernels.
ghstack-source-id: 114499399

Test Plan:
- Sandcastle CI
- Circle CI

Reviewed By: IvanKobzarev, ailzhang

Differential Revision: D24356769

fbshipit-source-id: 8660ca5be663fdc8985d9eb710ddaadbb43b0ddd
  • Loading branch information
xta0 authored and facebook-github-bot committed Oct 16, 2020
1 parent 8c629ec commit 5da4a08
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions c10/core/DispatchKeySet.h
Expand Up @@ -204,6 +204,7 @@ constexpr DispatchKeySet autogradother_backends = DispatchKeySet({
DispatchKey::FPGA,
DispatchKey::MSNPU,
DispatchKey::Vulkan,
DispatchKey::Metal,
DispatchKey::MKLDNN,
DispatchKey::OpenGL,
DispatchKey::OpenCL,
Expand Down

0 comments on commit 5da4a08

Please sign in to comment.