pytorch
97e8dcba - Fix mis-specified device arg name (#69645)

Commit
2 years ago
Fix mis-specified device arg name (#69645) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/69645 As noted in code comment: existing device operator is registered with input name `a`, which prevents torch.device(type="cuda") from working. add shim-layer here Test Plan: Imported from OSS Reviewed By: jbschlosser Differential Revision: D33515231 Pulled By: eellison fbshipit-source-id: c04af8158a9568a20cd5fbbbd573f6efab98fd60
Author
Elias Ellison
Parents
Loading