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.

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

Prepare for GPU support #22

Draft
wants to merge 28 commits into
base: main
Choose a base branch
from
Draft
Changes from 1 commit
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
43a645b
outline needed functions
tyler-a-cox Aug 9, 2024
0827abd
fill out gpu beam function
tyler-a-cox Aug 10, 2024
3c6a4ab
restructure simulation function for GPU
tyler-a-cox Aug 11, 2024
7663998
fix call to CPU beam interpolation
tyler-a-cox Aug 11, 2024
3ac59c8
add GPU/CPU structure
tyler-a-cox Aug 21, 2024
23fb2c3
remove coordinate transform file
tyler-a-cox Sep 10, 2024
fa5f12d
Merge branch 'main' into prepare_for_gpu
tyler-a-cox Sep 10, 2024
40cb821
add wrapper methods
tyler-a-cox Sep 13, 2024
0e5dffc
add memory calculations to utils.py
tyler-a-cox Sep 18, 2024
3d9aad5
set cpu structure
tyler-a-cox Sep 18, 2024
0bfc715
revert simulate.py
tyler-a-cox Sep 18, 2024
60bd9fc
fix inputs for cpu.simulate
tyler-a-cox Sep 18, 2024
172e53e
add wrapper.py for new kwargs
tyler-a-cox Sep 18, 2024
992f366
reorganize cpu and gpu functions
tyler-a-cox Sep 26, 2024
41dfd8f
run frequency loop in wrapper function
tyler-a-cox Sep 26, 2024
c5fc420
update documentation for GPU installation
tyler-a-cox Sep 26, 2024
aec3b20
remove old beam modules
tyler-a-cox Sep 27, 2024
9ae919a
add FFT modules
tyler-a-cox Sep 27, 2024
5ce0365
rename finufft files in cpu/gpu
tyler-a-cox Sep 27, 2024
2bcb825
precommit changes
tyler-a-cox Sep 27, 2024
d64d357
update with new simulation format
tyler-a-cox Sep 27, 2024
cc1f31a
Correct EarthLocation parameter
tyler-a-cox Sep 27, 2024
0109b34
update wrapper function
tyler-a-cox Sep 27, 2024
30ebeb6
update init to check for gpu first
tyler-a-cox Sep 28, 2024
23f1b7d
add coherency matrix product
tyler-a-cox Oct 1, 2024
53a59a3
underscore unsued parameters
tyler-a-cox Oct 1, 2024
774e169
fix nufft calls
tyler-a-cox Oct 1, 2024
0d6e68e
new file to supplement matvis coordinate rotations
tyler-a-cox Oct 1, 2024
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
Prev Previous commit
Next Next commit
update documentation for GPU installation
  • Loading branch information
tyler-a-cox committed Sep 26, 2024
commit c5fc420d91a2579800a93bcb975bb5b89e083e6b
22 changes: 21 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -14,7 +14,6 @@
## Limitations
- Currently no support for per-antenna beams
- Currently no support for polarized sky emission
- Currently no GPU support
- Diffuse sky models must be pixelized

## Installation
@@ -25,6 +24,27 @@ You can install `fftvis` via pip:
pip install fftvis
```

### Installation for GPU

Clone `finufft` repository
```
git clone https://github.com/flatironinstitute/finufft
cd finufft
```

Build `finufft` for `CUDA`
```bash
mkdir build
cd build
cmake -D FINUFFT_USE_CUDA=ON ..
cmake --build .
```

Install `Python` installation
```bash
pip install python/cufinufft
```

## Contributing
Contributions to `fftvis` are welcome! If you find any issues, have feature requests, or want to contribute improvements, please open an issue or submit a pull request on the GitHub repository: `fftvis` on GitHub