Skip to content

Optimal comm

Latest
Latest
Compare
Choose a tag to compare
@mloubout mloubout released this 10 Aug 18:15
· 12 commits to main since this release
0.3
b532554
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Uses restrictions bewtween each 1D DFT to only communicate the non-zero part of the fourier modes.

What's Changed

  • Small fixes for bench.py --device=cpu by @Infinoid in #4
  • Restrict FFTs in DistributedFNOBlock by @Infinoid in #3

New Contributors

Full Changelog: 0.2...0.3

Contributors

Infinoid
Assets 2
Loading