Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add lru_cache to _calculate_amat_eigenvecs to speed up pswf_operator #10

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

Add lru_cache to _calculate_amat_eigenvecs to speed up pswf_operator #10

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
6751481
Update version number to match pypi
tyler-a-cox Jul 1, 2022
a3c3ba9
Merge branch 'main' of https://github.com/HERA-Team/hera_filters
tyler-a-cox Jul 13, 2022
ffb262c
Merge branch 'main' of https://github.com/HERA-Team/hera_filters
tyler-a-cox Sep 7, 2022
e8c4c4f
Merge branch 'main' of https://github.com/HERA-Team/hera_filters
tyler-a-cox Oct 20, 2022
6b68c9f
Merge branch 'main' of https://github.com/HERA-Team/hera_filters
tyler-a-cox Feb 8, 2023
748cae0
Add lru_cache to speed up pswf computation
tyler-a-cox Feb 11, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions hera_filters/dspec.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
from scipy.optimize import leastsq, lsq_linear
import copy
from scipy import linalg
from functools import lru_cache

#DEFAULT PARAMETERS FOR CLEANs
CLEAN_DEFAULTS_1D={'tol':1e-9, 'window':'none',
Expand Down Expand Up @@ -2198,7 +2199,8 @@ def fit_solution_matrix(weights, design_matrix, cache=None, hash_decimal=10, fit
cache[opkey] = None
return cache[opkey]

def _calculate_amat(k, c):
@lru_cache(maxsize=8)
def _calculate_amat_eigenvecs(k, c):
"""
Computes the eigenvalues and eigenvectors of the decay coefficient matrix
used to compute prolate spheroidal wave functions. More information can be
Expand Down Expand Up @@ -2363,7 +2365,7 @@ def pswf_operator(
for fn, (fw, fc) in enumerate(zip(filter_half_widths, filter_centers)):
# Compute eigenvectors
c = fw * np.pi * (xmax - xmin)
eigenvals, eigenvecs = _calculate_amat(kmax, c)
eigenvals, eigenvecs = _calculate_amat_eigenvecs(kmax, c)

# Sort eigenvectors by largest eigenvalue
idx = np.argsort(eigenvals)
Expand Down