Skip to content

[RuntimeError] copy_() : CUDA error: unspecified launch failure #35225

@anjali411

Description

@anjali411

CUDA error: unspecified launch failure when I changed the dispatch here

AT_DISPATCH_ALL_TYPES_AND3(kHalf, kBool, kBFloat16, iter.dtype(0), "copy_", [&] {
to AT_DISPATCH_ALL_TYPES_AND_COMPLEX_AND3
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

cc @ezyang @anjali411 @dylanbespalko @ngimel

Metadata

Metadata

Assignees

Labels

module: complexRelated to complex number support in PyTorchmodule: cudaRelated to torch.cuda, and CUDA support in generaltriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions