Code

Fix halo again#23

Code

Merged

ASKabalan merged 8 commits intomainDifferentiableUniverseInitiative/jaxDecomp:mainfrom fix_halo_againDifferentiableUniverseInitiative/jaxDecomp:fix_halo_againCopy head branch name to clipboardJul 19, 2024

+71-16

Commits

Commits on Jul 5, 2024

• fix : Reduce halo, it should be half of the halo size

  ASKabalan

  committed
  52f7afe

• Allocate using cuda for halo_exchange by default ..allow fallback to XLA

  ASKabalan

  committed
  0ae6f56

• Clean up all GridDescriptor and halo_exchange memory at finalize

  ASKabalan

  committed
  e3d4c69

• Do not allocate with XLA if not selected

  ASKabalan

  committed
  8ca8f36

Commits on Jul 8, 2024

• Merge branch 'main' into fix_halo_again

  EiffL

  authored
  82f1395

Commits on Jul 19, 2024

• set worksize correctly when using Halo

  ASKabalan

  committed
  75e7032

• revert cuda allocation for halo, only XLA is done

  ASKabalan

  committed
  d59c638

• Merge remote-tracking branch 'upstream/main' into fix_halo_again

  ASKabalan

  committed
  81a4095