-
Notifications
You must be signed in to change notification settings - Fork 27.3k
Closed
Labels
module: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generaltriagedThis 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
CUDA error: unspecified launch failure when I changed the dispatch here
pytorch/aten/src/ATen/native/cuda/Copy.cu
Line 64 in 9441c7a
| AT_DISPATCH_ALL_TYPES_AND3(kHalf, kBool, kBFloat16, iter.dtype(0), "copy_", [&] { |
and ran this code on cuda:
import torch
a=torch.randn(2,3,dtype=torch.complex64).cuda()
b=a.to(torch.complex128)
RuntimeError: CUDA error: unspecified launch failure
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
module: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generaltriagedThis 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