Avatar for the GalacticDynamics user
GalacticDynamics
galax
BlogDocsChangelog

✨ feat(dynamics): integrate orbit with AbstractOrbitField

#724Merged
Comparing
nstarman:gd/compute_orbit
(
c062dd5
) with
main
(
7e02883
)
CodSpeed Performance Gauge
0%
Improvements
0
Regressions
0
Untouched
20
New
6
Dropped
0
Ignored
0

Benchmarks

New

test_execute[integrate_orbit-field-('HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-field-('HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
5.8 ms
test_execute[integrate_orbit-field-Scan-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-field-Scan-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
5.7 ms
test_execute[integrate_orbit-field-VMap-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-field-VMap-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
5.8 ms
test_jit_compile[integrate_orbit-field-('HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-field-('HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
1.1 ms
test_jit_compile[integrate_orbit-field-Scan-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-field-Scan-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
1.1 ms
test_jit_compile[integrate_orbit-field-VMap-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-field-VMap-('type', 'HamiltonianField', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
N/A
N/A
1.1 ms

Passed

test_jit_compile[integrate_orbit-scalar-('HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-scalar-('HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
+1%
1.1 ms
1.1 ms
test_jit_compile[simulate_stream-vmap-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_jit_compile[simulate_stream-vmap-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
+1%
1.3 ms
1.3 ms
test_jit_compile[simulate_stream-scan-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_jit_compile[simulate_stream-scan-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
1.3 ms
1.3 ms
test_jit_compile[integrate_orbit-scalar-Scan-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-scalar-Scan-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
1.1 ms
1.1 ms
test_jit_compile[integrate_orbit-scalar-NoLoop-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-scalar-NoLoop-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
1.1 ms
1.1 ms
test_execute[integrate_orbit-scalar-NoLoop-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-scalar-NoLoop-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
5.7 ms
5.7 ms
test_execute[integrate_orbit-scalar-('HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-scalar-('HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
5.7 ms
5.7 ms
test_execute[integrate_orbit-scalar-VMap-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-scalar-VMap-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
5.8 ms
5.7 ms
test_execute[simulate_stream-scan-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_execute[simulate_stream-scan-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
21.8 s
21.7 s
test_jit_compile[simulate_stream-determine-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_jit_compile[simulate_stream-determine-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
1.3 ms
1.3 ms
test_jit_compile[simulate_stream-no-flag-('HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_jit_compile[simulate_stream-no-flag-('HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
1.3 ms
1.3 ms
test_execute[simulate_stream-no-flag-('HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_execute[simulate_stream-no-flag-('HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
20.4 s
20.4 s
test_execute[simulate_stream-vmap-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_execute[simulate_stream-vmap-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
20.4 s
20.4 s
test_execute[simulate_stream-determine-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_execute[simulate_stream-determine-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
20.4 s
20.4 s
test_execute[simulate_stream-df-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_execute[simulate_stream-df-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
20.4 s
20.4 s
test_execute[integrate_orbit-scalar-Scan-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_execute[integrate_orbit-scalar-Scan-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
5.7 ms
5.7 ms
test_jit_compile[sample-Fardal15-('PRNGKeyArray', 'HernquistPotential', 'int', 'ArrayImpl', 'ArrayImpl', 'float')]
tests/benchmark/test_experimental.py::test_jit_compile[sample-Fardal15-('PRNGKeyArray', 'HernquistPotential', 'int', 'ArrayImpl', 'ArrayImpl', 'float')]
CodSpeed Performance Gauge
0%
1.3 ms
1.3 ms
test_jit_compile[integrate_orbit-scalar-VMap-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
tests/benchmark/test_experimental.py::test_jit_compile[integrate_orbit-scalar-VMap-('type', 'HernquistPotential', 'tuple', 'ArrayImpl', 'ArrayImpl')]
CodSpeed Performance Gauge
0%
1.1 ms
1.1 ms
test_jit_compile[simulate_stream-df-('type', 'HernquistPotential', 'tuple')]
tests/benchmark/test_experimental.py::test_jit_compile[simulate_stream-df-('type', 'HernquistPotential', 'tuple')]
CodSpeed Performance Gauge
0%
3.8 ms
3.9 ms
test_execute[sample-Fardal15-('PRNGKeyArray', 'HernquistPotential', 'int', 'ArrayImpl', 'ArrayImpl', 'float')]
tests/benchmark/test_experimental.py::test_execute[sample-Fardal15-('PRNGKeyArray', 'HernquistPotential', 'int', 'ArrayImpl', 'ArrayImpl', 'float')]
CodSpeed Performance Gauge
-1%
601 µs
608.3 µs

Commits

Click on a commit to change the comparison range
Base
main
7e02883
0%
✅ test(dynamics): benchmark orbitfield arg Signed-off-by: Nathaniel Starkman <nstarman@users.noreply.github.com>
c062dd5
5 days ago
by nstarman
Home Terms PrivacyDocs