"extensions/csrc/cuda/layer_norm_cuda.cpp" did not exist on "58580b50fec5c58a575e589428bddeee18408003"
- 11 Mar, 2022 40 commits
-
-
Jiarui Fang authored
-
Jiarui Fang authored
* place params on cpu after zero init context * polish code
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
Jiarui Fang authored
-
Jiarui Fang authored
-
HELSON authored
-
HELSON authored
-
ver217 authored
-
Jiarui Fang authored
-
HELSON authored
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
jiaruifang authored
-
Frank Lee authored
-
Frank Lee authored
-
Jie Zhu authored
* fix merge conflict modify unit test remove unnessesary log info reformat file * remove unused module * remove unnecessary sync function * change doc string style from Google to Sphinx
-
ver217 authored
-
jiaruifang authored
-
jiaruifang authored
-
Jiarui Fang authored
-
Frank Lee authored
-
HELSON authored
-
Jiarui Fang authored
* add zero init context * add more flags for zero init context fix bug of repeated converting param to ShardedParamV2 * polish code
-
1SAA authored
Fixed bug for learning rate scheduler
-
binmakeswell authored
-
LuGY authored
* Added CPU Adam * finished the cpu adam * updated the license * delete useless parameters, removed resnet * modified the method off cpu adam unittest * deleted some useless codes * removed useless codes Co-authored-by:
ver217 <lhx0217@gmail.com> Co-authored-by:
Frank Lee <somerlee.9@gmail.com> Co-authored-by:
jiaruifang <fangjiarui123@gmail.com>
-
Jiarui Fang authored
-
Jiarui Fang authored
* init shard param from shape tuple * add more unitest for shard param * add set_payload method for ShardedParam * [zero] add shareded tensor class * polish code * add shard stratgy * move shard and gather logic to shard strategy from shard tensor. * polish code
-
ver217 authored
-
ver217 authored
-
ver217 authored
-
Jiarui Fang authored
-