Skip to content
GitLab
Explore
Sign in
Register
wanggh
apex
Repository
apex
apex
transformer
tensor_parallel
layers.py
Find file
Blame
History
Permalink
ColumnParallelLinearWithAsyncAllreduce autocast support (#1183)
· b3da6036
Masaki Kozuki
authored
Oct 06, 2021
* [ColumnParallelLinear] Test behavior in autocast * fix test * casts manually to autocast dtype
b3da6036