graphcore-research / pyscf-ipu

PySCF on IPU
https://github.com/graphcore-research/pyscf-ipu#readme
Apache License 2.0
42 stars 2 forks source link

Add function decorator for comparing fp64 to fp32 #110

Closed hatemhelal closed 1 year ago

hatemhelal commented 1 year ago

This adds a function decorator that runs a function first with all inputs cast to fp32 and a second time with fp64. Following this, max |fp64 - fp32| is printed for each of the function outputs.