Skip to content
GitLab
Explore
Sign in
Register
wanggh
apex
Repository
Branches
Overview
Active
Stale
All
asp_permutation_extension
55b6f83d
·
[Fix] Track the permutation sequence for progressive channel swap strategy.
·
Nov 10, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
master
default
25bfcb91
·
conditionally import amp_C (#1211)
·
Nov 09, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
thc_headers
05f2d96b
·
`FastLayerNorm` compat with `autocast` (#1203)
·
Oct 27, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
optional_caller_supplied_communicator
e777bddb
·
Optional NCCL communicator argument to init method
·
Sep 02, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
bug_fix_in_wgrad
9e295728
·
Bug fix in wgrad
·
Sep 02, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
more_bug_fixes
8c4a0075
·
Revert some changes
·
Sep 02, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
bug_fixes
67a0ffcb
·
More detailed output
·
Sep 02, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
rework_spatial_bottleneck_code_split
b6980a0d
·
Add functions to compute grad_out1, grad_out1_halo
·
Sep 01, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
deyuf/bottleneck_block
e51bc22c
·
sync cudnn-frontend
·
Apr 16, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
mha_additive
21565099
·
fix mha test
·
Jan 20, 2021
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
kyu/dist_adam
4ad53bf8
·
v4 -> v1
·
Oct 19, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
kyu/dist_lamb_fix
4ad53bf8
·
v4 -> v1
·
Oct 19, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
fix_imagenet_example
9637fc79
·
use reshape instead of view
·
Sep 28, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
fix_syncbn_sizes
729b33b8
·
Should pass stricter stride/size checks in pytorch
·
Aug 14, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
sm80_fix
fc35df8a
·
move sm80 code inside MHA
·
Aug 07, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
sm80_test
71d1d919
·
add sm80 for CUDA >= 11 instead of == 11
·
Jul 31, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
sbn_fix
8060cf3f
·
cleaning
·
Jul 02, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
fix__div__
c2b61c5f
·
syntax
·
Jun 30, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
distopt_flatten_option
b87fcb4d
·
Bug fix
·
Jun 12, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
schetlur/stream_bug_fix
060bd5cd
·
Update softmax.h
·
Jun 11, 2020
Select Archive Format
Download source code
zip
tar.gz
tar.bz2
tar
Prev
1
2
3
4
5
6
Next