HIPRTC_ERROR_INTERNAL_ERROR while attempting tensor.abs().pow(2.0)
Closed this issue · 1 comments
AznamirWoW commented
Code
import torch
dims = (1)
real_part = torch.randn(dims, dtype=torch.float64)
imaginary_part = torch.randn(dims, dtype=torch.float64)
#on cpu
complex_tensor = torch.complex(real_part, imaginary_part).to("cpu")
complex_tensor1 = complex_tensor.abs()
#on cuda
complex_tensor = torch.complex(real_part, imaginary_part).to("cuda")
complex_tensor2 = complex_tensor.abs()
lshqqytiger commented
Closing as this issue is impossible to solve unless you build PyTorch yourself.