Closed gautierronan closed 1 year ago
Hi @gautierronan,
Currently, hidet does not support complex-valued tensors. Can I know what networks are you working on that need complex-valued tensors?
Yes, we are working on a library for the simulation of quantum systems (https://github.com/PierreGuilmin/torchqdynamics, but still WIP), and all quantum objects are complex-valued by nature. I was looking at hidet for compilation and for its ability to define custom tensor operations.
Overall, the whole quantum physics community (industry and academia, quantum computing and other fields) would really benefit from that.
Hi @gautierronan,
I added the complex64 and complex128 data types in this PR https://github.com/hidet-org/hidet/pull/200. But I am still not quite sure what operators or operations in the kernel you need. Please let us know if you have any questions in using hidet. You are also welcome to contribute to the complex type support if you find anything missing here, and I would like to help.
As the title says. Thanks !