In the core code currently there is an unused preprocessor flag to "USE_FLOATS" which is a remnant of the legacy code (https://github.com/amnsbr/bnm_cuda). After making sure it works with the updated code of cuBNM, we might be able to use this as an installation option which can be controlled by setting an environment variable before pip install or by using pip install cubnm[float].
However, ideally we should be able to make this a runtime option so that it doesn't require a separate installation, but I can imagine that this can be very challenging.
It is also important to think if this option should really exist, and if it is really okay to use float precision in some simulations.
In the core code currently there is an unused preprocessor flag to "USE_FLOATS" which is a remnant of the legacy code (https://github.com/amnsbr/bnm_cuda). After making sure it works with the updated code of cuBNM, we might be able to use this as an installation option which can be controlled by setting an environment variable before
pip installor by usingpip install cubnm[float].However, ideally we should be able to make this a runtime option so that it doesn't require a separate installation, but I can imagine that this can be very challenging.
It is also important to think if this option should really exist, and if it is really okay to use float precision in some simulations.