Skip to content

GitLab

Switch branch/tag
  1. 28 Nov, 2023 1 commit
  3. 25 Jan, 2021 1 commit
    • Jeff Daily's avatar
      fix bugs in syncbn (#46) · 3f49dbf0
      Jeff Daily authored 
      - incorrect use of __shfl_down
- fix warp size assumptions
- update unit tests to exit on failure
      3f49dbf0
  5. 06 Jul, 2020 1 commit
    • jjsjann123's avatar
      [sync BN] (#792) · 1ff54b8f
      jjsjann123 authored 
      * [sync BN]

support non-uniform batch size across process group.

TODO: test should be added once cleaned up.

* updating unit tests

* new unit tests for different inputs

* cleaning
      1ff54b8f
  7. 06 Nov, 2019 1 commit
  9. 26 Jul, 2019 1 commit
    • jjsjann123's avatar
      [sbn update] (#384) · 896ecdd6
      jjsjann123 authored 
      fixing empty return from python implementation
  adding proper test to verify functional correctness for python implementation
      896ecdd6
  11. 12 Jul, 2019 1 commit
    • jjsjann123's avatar
      [sbn update] (#384) · 574fe244
      jjsjann123 authored 
      fixing empty return from python implementation
  adding proper test to verify functional correctness for python implementation
      574fe244
  13. 12 Mar, 2019 1 commit
  15. 26 Feb, 2019 1 commit