I’m sorry, I have quite a few questions. There is an error when I try to run the following code:
import torch
import torch_mlir
class FFT(torch.nn.Module):
def forward(self, z1, z2):
temp=z1+z2
return temp
z1 = torch.tensor([1+2j], dtype=torch.complex64)
z2 = torch.tensor([5-2j], dtype=torch.complex64)
compiled = torch_mlir.compile(FFT(), (z1,z2), output_type=torch_mlir.OutputType.LINALG_ON_TENSORS)
print(compiled)
The torch_mlir.complie function will be aborted, “Aborted” will be print in terminal on Mac, and there isn’t anything printed in terminal on Windows.