Commits · 3a0eb3e69f356ec048c429347aa705db9c369680 · OpenDAS / FastMoE

19 May, 2023 1 commit
- changed .float() to .as_type(self.weight) · 1f7c04c9
  Arturo Ghinassi authored May 19, 2023
```
AMP support
```
  1f7c04c9
18 May, 2023 1 commit
- convert input to same type as weight for mixed precision training · 2196b766
  Arturo Ghinassi authored May 18, 2023
```
When using CUDA AMP FMoE Linear throws type error as input is half() and weights are float()
```
  2196b766
10 Oct, 2021 1 commit
- tide up fmoe python code · d9dba929
  Rick Ho authored Oct 10, 2021
  
  d9dba929