pytorch
66158a6e - Enable AutogradXPU DispatchKey for Intel heterogeneous computation platform. (#61105)

Commit
3 years ago
Enable AutogradXPU DispatchKey for Intel heterogeneous computation platform. (#61105) Summary: Add string wrapper for AutogradXPU to enable this DispatchKey. We are going to use AutogradXPU as custom autograd backend, which needs this DispatchKey. This sting wrapper is used to map AutogradXPU to the corresponding DispatchKey. Pull Request resolved: https://github.com/pytorch/pytorch/pull/61105 Reviewed By: malfet Differential Revision: D29557697 Pulled By: ezyang fbshipit-source-id: f0c8155decc8e2fd90741650a05de5a8b5a70121
Author
Parents
Loading