-
Notifications
You must be signed in to change notification settings - Fork 25.6k
Closed
Labels
module: bfloat16module: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: randomRelated to random number generation in PyTorch (rng generator)Related to random number generation in PyTorch (rng generator)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
This is a known issue, that requires further investigation, currently calling any random_() method on bfloat16 CUDA tensor on Windows makes CUDA context invalid, all subsequent CUDA calls fail with 'CUDA error: unspecified launch failure'
Assignee, please look for "TODO: https://github.com/pytorch/pytorch/issues/33793" in the source code
cc @ngimel
Metadata
Metadata
Assignees
Labels
module: bfloat16module: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generalmodule: randomRelated to random number generation in PyTorch (rng generator)Related to random number generation in PyTorch (rng generator)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module