Commits · 3a41edb8c29eae02fd286302ab85d809c73fff18 · OpenDAS / FastMoE

11 Sep, 2023 1 commit
- use torchstream everywhere · 4f9f77f8
  Rick Ho authored Sep 11, 2023
  
  4f9f77f8
30 Sep, 2021 2 commits
- Update local_exchange.cuh · 976ef2b8
  hclearner authored Sep 30, 2021
  
  976ef2b8
- add fastmoe support rocm4.0.1 · 2a77b772
  huchen1 authored Sep 30, 2021
  
  2a77b772
16 Jun, 2021 1 commit

Rick Ho authored Jun 16, 2021

* use single variable instead of vector in c functions

* expert count kernel

* remove all lists

* fix old tests

295a615a

19 May, 2021 1 commit
- remove local scatter and gather in cuda · 52e57316
  Rick Ho authored May 19, 2021
  
  52e57316
18 May, 2021 1 commit
- use customized pos assignment kernel to support -1 · 980cf4b6
  Rick Ho authored May 18, 2021
  
  980cf4b6
26 Apr, 2021 1 commit
- reconstruct fmoe cuda code · bb3c8966
  Rick Ho authored Apr 26, 2021
  
  bb3c8966