fastlisaresponse 1.0.9__cp312-cp312-macosx_11_0_arm64.whl

fastlisaresponse/utils/__init__.py

@@ -0,0 +1 @@
+from .utility import get_overlap

fastlisaresponse/utils/utility.py

@@ -0,0 +1,82 @@
+import numpy as np
+
+try:
+    import cupy as cp
+    from pyresponse import get_response_wrap as get_response_wrap_gpu
+    from pyresponse import get_tdi_delays_wrap as get_tdi_delays_wrap_gpu
+
+    gpu = True
+
+except (ImportError, ModuleNotFoundError) as e:
+    import numpy as xp
+
+    gpu = False
+
+
+def get_overlap(sig1, sig2, phase_maximize=False, use_gpu=False):
+    """Calculate the mismatch across TDI channels
+
+    Calculates the overlap between two sets of TDI observables in the time
+    domain. The overlap is complex allowing for the addition of overlap
+    over all channels. It can be phase maximized as well.
+
+    This function has GPU capabilities.
+
+    Args:
+        sig1 (list or xp.ndarray): TDI observables for first signal. Must be ``list`` of
+            ``xp.ndarray`` or a single ``xp.ndarray``. Must have same length as ``sig2`` in terms
+            of number of channels and length of the indivudal channels.
+        sig2 (list or xp.ndarray): TDI observables for second signal. Must be ``list`` of
+            ``xp.ndarray`` or a single ``xp.ndarray``. Must have same length as ``sig1`` in terms
+            of number of channels and length of the individual channels.
+        phase_maximize (bool, optional): If ``True``, maximize over the phase in the overlap.
+            This is equivalent to getting the magnitude of the phasor that is the complex
+            overlap. (Defaut: ``False``)
+        use_gpu (bool, optional): If ``True``, use the GPU. This sets ``xp=cupy``. If ``False,
+            use the CPU and set ``xp=numpy``.
+
+    Returns:
+        double: Overlap as a real value.
+
+    """
+
+    # choose right array library
+    if use_gpu:
+        xp = cp
+    else:
+        xp = np
+
+    # check inputs
+    if not isinstance(sig1, list):
+        if not isinstance(sig1, xp.ndarray):
+            raise ValueError("sig1 must be list of or single xp.ndarray.")
+
+        elif sig1.ndim < 2:
+            sig1 = [sig1]
+
+    if not isinstance(sig2, list):
+        if not isinstance(sig2, xp.ndarray):
+            raise ValueError("sig1 must be list of or single xp.ndarray.")
+
+        elif sig1.ndim < 2:
+            sig2 = [sig2]
+
+    assert len(sig1) == len(sig2)
+    assert len(sig1[0]) == len(sig2[0])
+
+    # complex overlap
+    overlap = 0.0 + 1j * 0.0
+    for sig1_i, sig2_i in zip(sig1, sig2):
+        overlap_i = np.dot(np.fft.rfft(sig1_i).conj(), np.fft.rfft(sig2_i)) / np.sqrt(
+            np.dot(np.fft.rfft(sig1_i).conj(), np.fft.rfft(sig1_i))
+            * np.dot(np.fft.rfft(sig2_i).conj(), np.fft.rfft(sig2_i))
+        )
+        overlap += overlap_i
+
+    overlap /= len(sig1)
+
+    if phase_maximize:
+        return np.abs(overlap)
+
+    else:
+        return overlap.real

fastlisaresponse-1.0.9.dist-info/METADATA

@@ -0,0 +1,127 @@
+Metadata-Version: 2.1
+Name: fastlisaresponse
+Version: 1.0.9
+Home-page: https://github.com/mikekatz04/lisa-on-gpu
+Author: Michael Katz
+Author-email: mikekatz04@gmail.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: GNU General Public License (GPL)
+Classifier: Environment :: GPU :: NVIDIA CUDA
+Classifier: Natural Language :: English
+Classifier: Programming Language :: C++
+Classifier: Programming Language :: Cython
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+
+# fastlisaresponse: Generic LISA response function for GPUs
+
+This code base provides a GPU-accelerated version of the generic time-domain LISA response function. The GPU-acceleration allows this code to be used directly in Parameter Estimation.
+
+Please see the [documentation](https://mikekatz04.github.io/lisa-on-gpu/) for further information on these modules. The code can be found on Github [here](https://github.com/mikekatz04/lisa-on-gpu). It can be found on [Zenodo](https://zenodo.org/record/3981654#.XzS_KRNKjlw).
+
+If you use all or any parts of this code, please cite [arXiv:2204.06633](https://arxiv.org/abs/2204.06633). See the [documentation](https://mikekatz04.github.io/lisa-on-gpu/) to properly cite specific modules.
+
+
+## Getting Started
+
+Install with pip (CPU only for now):
+```
+pip install fastlisaresponse
+```
+To import fastlisaresponse:
+
+```
+from fastlisaresponse import ResponseWrapper
+```
+
+See [examples notebook](https://github.com/mikekatz04/lisa-on-gpu/blob/master/examples/fast_LISA_response_tutorial.ipynb).
+
+
+### Prerequisites
+
+Now (version 1.0.9) `fastlisaresponse` requires the newest version of [LISA Analysis Tools](github.com/mikekatz04/LISAanalysistools). You can run `pip install lisaanalysistools`.
+
+To install this software for CPU usage, you need Python >3.4 and NumPy. To run the examples, you will also need jupyter and matplotlib. We generally recommend installing everything, including gcc and g++ compilers, in the conda environment as is shown in the examples here. This generally helps avoid compilation and linking issues. If you use your own chosen compiler, you will need to make sure all necessary information is passed to the setup command (see below). You also may need to add information to the `setup.py` file.
+
+To install this software for use with NVIDIA GPUs (compute capability >2.0), you need the [CUDA toolkit](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html) and [CuPy](https://cupy.chainer.org/). The CUDA toolkit must have cuda version >8.0. Be sure to properly install CuPy within the correct CUDA toolkit version. Make sure the nvcc binary is on `$PATH` or set it as the `CUDAHOME` environment variable.
+
+
+### Installing
+
+
+Install with pip (CPU only for now):
+```
+pip install fastlisaresponse
+```
+
+To install from source:
+
+0) [Install Anaconda](https://docs.anaconda.com/anaconda/install/) if you do not have it.
+
+1) Create a virtual environment.
+
+```
+conda create -n lisa_resp_env -c conda-forge gcc_linux-64 gxx_linux-64 numpy Cython scipy jupyter ipython h5py matplotlib python=3.12
+conda activate lisa_resp_env
+```
+
+    If on MACOSX, substitute `gcc_linux-64` and `gxx_linus-64` with `clang_osx-64` and `clangxx_osx-64`.
+
+    If you want a faster install, you can install the python packages (numpy, Cython, scipy, tqdm, jupyter, ipython, h5py, requests, matplotlib) with pip.
+
+2) Clone the repository.
+
+```
+git clone https://github.com/mikekatz04/lisa-on-gpu.git
+cd lisa-on-gpu
+```
+
+3) If using GPUs, use pip to [install cupy](https://docs-cupy.chainer.org/en/stable/install.html). 
+
+```
+pip install cupy-12x
+```
+
+4) Run install. Make sure CUDA is on your PATH.
+
+```
+python scripts/prebuild.py
+pip install .
+```
+
+## Running the Tests
+
+Run the example notebook or the tests using `unittest` from the main directory of the code:
+```
+python -m unittest discover
+```
+
+## Contributing
+
+Please read [CONTRIBUTING.md](CONTRIBUTING.md) for details on our code of conduct, and the process for submitting pull requests to us.
+
+## Versioning
+
+We use [SemVer](http://semver.org/) for versioning. For the versions available, see the [tags on this repository](https://github.com/mikekatz04/lisa-on-gpu/tags).
+
+Current Version: 1.0.9
+
+## Authors
+
+* **Michael Katz**
+* Jean-Baptiste Bayle
+* Alvin J. K. Chua
+* Michele Vallisneri
+
+### Contibutors
+
+* Maybe you!
+
+## License
+
+This project is licensed under the GNU License - see the [LICENSE.md](LICENSE.md) file for details.
+
+## Acknowledgments
+
+* It was also supported in part through the computational resources and staff contributions provided for the Quest/Grail high performance computing facility at Northwestern University.

fastlisaresponse-1.0.9.dist-info/RECORD

@@ -0,0 +1,17 @@
+fastlisaresponse/__init__.py,sha256=wwmiIBy9IuFwoc4jQyJVkJBhjB8B1XZjerTe_E8FkI8,53
+fastlisaresponse/_version.py,sha256=q2ACMkQx0Hm5xKo2YKJFBE7rTruciTSIN6AvWWL9nB8,22
+fastlisaresponse/pointer_adjust.py,sha256=TjcSehyffLxwgJnrAmcFlPvxXb3XPElMoHXLBOQN-PI,736
+fastlisaresponse/response.py,sha256=_dHVDuSTbUhxnUW_gmweAJ0M-jLgPZMvZpVQmkx2eiE,28665
+fastlisaresponse/cutils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastlisaresponse/cutils/pyresponse_cpu.cpython-312-darwin.so,sha256=lO19JH-noRl8llH7by9hkryurZ9RKsHBLKz0LMM5CCk,126136
+fastlisaresponse/cutils/include/LISAResponse.hh,sha256=mbWEq31Q98PR7XCzhb6SqmygJrXGcQK_fs3sHbmZNe8,1024
+fastlisaresponse/cutils/include/cuda_complex.hpp,sha256=qjBjBf5ctvcKYtMPO2ngqTi6qgq6xsnalFwaZG_NLTw,37562
+fastlisaresponse/cutils/src/LISAResponse.cpp,sha256=AfIdBpwTVtwtpL0Rf2p6WB_XG4YZhII6_z66ldedtuE,29994
+fastlisaresponse/cutils/src/LISAResponse.cu,sha256=AfIdBpwTVtwtpL0Rf2p6WB_XG4YZhII6_z66ldedtuE,29994
+fastlisaresponse/cutils/src/responselisa.pyx,sha256=RexvhiGcb2fK_i1YIQMQxGZLKfJYukn59Hd3wOkZqHw,3075
+fastlisaresponse/utils/__init__.py,sha256=pf2NmWKs_uQNzlyA5iNO1gTRDISKNmIIsvOcKqQ3hgw,33
+fastlisaresponse/utils/utility.py,sha256=NrJdBmEnLkLPk6Ile1TZg8jNLw6xERiSp58iGVlz01s,2709
+fastlisaresponse-1.0.9.dist-info/METADATA,sha256=tuXk6U6D0TM0nR--57ONepNeRKzRX_LdiBYR2gf9M38,4727
+fastlisaresponse-1.0.9.dist-info/WHEEL,sha256=7Wd-yga4fjSiXpUH443rsPZpiZ4h8-uNrXJrYRW_e14,109
+fastlisaresponse-1.0.9.dist-info/top_level.txt,sha256=J4M7Xx_52RqYZrLf99ryAbQjq8GbVjgZdhxekkYw8lg,17
+fastlisaresponse-1.0.9.dist-info/RECORD,,

fastlisaresponse-1.0.9.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.6.0)
+Root-Is-Purelib: false
+Tag: cp312-cp312-macosx_11_0_arm64
+

fastlisaresponse-1.0.9.dist-info/top_level.txt

@@ -0,0 +1 @@
+fastlisaresponse