pytorch
d3de3760 - Support fused_dropout with XPU backend (#60231)

Commit
3 years ago
Support fused_dropout with XPU backend (#60231) Summary: ## Motivation Enable the fused dropout optimization on XPU devices. ## Solution Add XPU device in the fused dropout acceptable checking. Pull Request resolved: https://github.com/pytorch/pytorch/pull/60231 Reviewed By: jbschlosser Differential Revision: D29437659 Pulled By: ezyang fbshipit-source-id: b77245bb53d3ac93ab30a2a85994376ae5928c34
Author
Parents
Loading