Closed beckermr closed 3 months ago
Comparing host-init-interp
(997f855) with main
(b6cbbcd)
⚡ 2
improvements
✅ 9
untouched benchmarks
Benchmark | main |
host-init-interp |
Change | |
---|---|---|---|---|
⚡ | test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile] |
1,558 ms | 492 ms | ×3.2 |
⚡ | test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile] |
261.5 ms | 160.2 ms | +63.22% |
This PR is a small refactor to move the init of the interpolant classes to the host. This pattern makes more sense in general since once we apply JIT, the constants computed on the host will get embeded into the JAX code and moved to the device with the code.