pytorch
1e2df9e4 - [cuda] masked_scatter : static_cast init_value to circumvent cuda 11.2 issue (#51614)

Commit
3 years ago
[cuda] masked_scatter : static_cast init_value to circumvent cuda 11.2 issue (#51614) Summary: Fixes https://github.com/pytorch/pytorch/issues/51544 Tested locally as there is no CI for 11.2 as of now. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51614 Reviewed By: malfet Differential Revision: D26253965 Pulled By: ngimel fbshipit-source-id: 6d666a54871510ad0d00f915e45bbebcebc93015
Author
Parents
Loading