search

hidet-org / hidet

An open-source efficient deep learning framework/compiler, written in python.
https://hidet.org
Apache License 2.0
648 stars 52 forks source link

[Question] Does hidet support the compilation of complex-valued tensors? #192

Closed gautierronan closed 1 year ago

gautierronan commented 1 year ago

As the title says. Thanks !

yaoyaoding commented 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?

gautierronan commented 1 year ago

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.

yaoyaoding commented 1 year ago

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.