Commits · ba3778f761fa17bbda702a37b04da79798db2637 · Wassim KABALAN / jaxDecomp

Oct 21, 2024
- add fft tests · ba3778f7
  Wassim KABALAN authored 4 months ago
  
  ba3778f7
- some cleanups in tests · e7c74097
  Wassim KABALAN authored 4 months ago
  
  e7c74097
- clean up fftfreq · 56575489
  Wassim KABALAN authored 4 months ago
  
  56575489
- use pybind locally without git submodule · 78a675bb
  Wassim KABALAN authored 4 months ago
  
  78a675bb
- use pybind as build dependency not gitmodule · 707138ad
  Wassim KABALAN authored 4 months ago
  
  707138ad
- format · 0eadb5be
  Wassim KABALAN authored 4 months ago
  
  0eadb5be
- Add test workflows · 5eea5130
  Wassim KABALAN authored 4 months ago
  
  5eea5130
- Remove prints · a8cb63ab
  Wassim KABALAN authored 4 months ago
  
  a8cb63ab
- format · a7f3bcf0
  Wassim KABALAN authored 4 months ago
  
  a7f3bcf0
- update transpose test · ed11ff03
  Wassim KABALAN authored 4 months ago
  
  ed11ff03
- Update Halo test · 354cb916
  Wassim KABALAN authored 4 months ago
  
  354cb916
- Update FFTs test · 3acebece
  Wassim KABALAN authored 4 months ago
  
  3acebece
- Some minor fixes in transpose · 1dd1aa60
  Wassim KABALAN authored 4 months ago
  
  1dd1aa60
- Remove strict dependency on global mesh for FFTs · 316ff4a9
  Wassim KABALAN authored 4 months ago
  
  316ff4a9
- Remove strict dependency on global mesh · bb9abd0d
  Wassim KABALAN authored 4 months ago
  
  bb9abd0d
- minor fix halo jax · a61a0d39
  Wassim KABALAN authored 4 months ago
  
  a61a0d39
- Halo is a two int tuple · e06a06e2
  Wassim KABALAN authored 4 months ago
  
  e06a06e2
- revert halo to tuple pdims · 6ce7cf08
  Wassim KABALAN authored 4 months ago
  
  6ce7cf08
Oct 19, 2024
- Increase test_transpose size to work on 8 devices · 1279fb9a
  Wassim KABALAN authored 4 months ago
  
  1279fb9a
- Mirror extra transpose change in fft test · 7df2e642
  Wassim KABALAN authored 4 months ago
  
  7df2e642
- test locally on 8 devices · f73b6fe2
  Wassim KABALAN authored 4 months ago
  
  f73b6fe2
- expose fftfreq3d · 3475a6fe
  Wassim KABALAN authored 4 months ago
  
  3475a6fe
- get_transpose_order will now need a local mesh · 7011b941
  Wassim KABALAN authored 4 months ago
  
  7011b941
- Implement FFTFreq · 5c122292
  Wassim KABALAN authored 4 months ago
  
  5c122292
- update Transposes to avoid using global mesh · 3466f480
  Wassim KABALAN authored 4 months ago
  
  3466f480
- update Halo to avoid using global mesh · 3f9773e7
  Wassim KABALAN authored 4 months ago
  
  3f9773e7
- update FFTs to avoid using global mesh · d3610332
  Wassim KABALAN authored 4 months ago
  
  d3610332
Oct 17, 2024
- [DRAFT] implementation of distributed fftfreq · b3635918
  Wassim KABALAN authored 4 months ago
  
  b3635918
- [DRAFT] implementation of distributed fftfreq · 92d21244
  Wassim KABALAN authored 4 months ago
  
  92d21244
- format · f0bc261d
  Wassim KABALAN authored 4 months ago
  
  f0bc261d
- expose get_output_specs · 3beb8a38
  Wassim KABALAN authored 4 months ago
  
  3beb8a38
- allow non periodic halo exchange · f26a6107
  Wassim KABALAN authored 4 months ago
  
  f26a6107
- remove prints · a0c02aba
  Wassim KABALAN authored 4 months ago
  
  a0c02aba
- format and add jaxtyping as dependency · 43b50c8b
  Wassim KABALAN authored 4 months ago
  
  43b50c8b
- expose FFTs · c7e548cf
  Wassim KABALAN authored 4 months ago
  
  c7e548cf
- allow testing and simulation distrubtion on CPUs · 91f4b85f
  Wassim KABALAN authored 4 months ago
  
  91f4b85f
- test transpose · 32d6b8d3
  Wassim KABALAN authored 4 months ago
  
  32d6b8d3
- test halo · 98922933
  Wassim KABALAN authored 4 months ago
  
  98922933
- clean up · 14401b8e
  Wassim KABALAN authored 4 months ago
  
  14401b8e
- factorize and implement jax Transpose · e6dc5473
  Wassim KABALAN authored 4 months ago
  
  e6dc5473

Admin message