master
apex
Repository badge with AMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
apex\contrib\index_mul_2d\index_mul_2d.py
25 1 -
A
apex\contrib\clip_grad\clip_grad.py
24 2 1
A
tests\L1\transformer\pipeline_parallel_fwd_bwd_ucc_async.py
24 1 2
A
apex\contrib\test\group_norm\test_group_norm.py
23 3 -
A
apex\parallel\optimized_sync_batchnorm_kernel.py
23 1 1
A
apex\contrib\sparsity\permutation_search_kernels\call_permutation_search_kernels.py
22 2 1
A
tests\L0\run_mlp\test_mlp.py
22 1 -
A
apex\contrib\csrc\layer_norm\ln_api.cpp
22 2 1
A
apex\contrib\csrc\fmha\fmha_api.cpp
21 3 3
A
apex\contrib\peer_memory\peer_memory.py
21 1 3
A
apex\contrib\multihead_attn\encdec_multihead_attn.py
21 2 -
A
apex\parallel\LARC.py
21 1 -
A
apex\contrib\csrc\fmha\src\fmha\gemm.h
20 1 7
A
tests\L0\run_amp\test_cache.py
20 2 -
A
csrc\type_shim.h
20 1 -
A
tests\L0\run_amp\test_multi_tensor_scale.py
20 1 -
A
apex\contrib\test\clip_grad\test_clip_grad.py
19 1 -
A
tests\L0\run_amp\test_update_scale_hysteresis.py
19 1 -
A
csrc\megatron\fused_rotary_positional_embedding.h
19 3 -
A
apex\contrib\test\optimizers\test_distributed_fused_lamb.py
19 2 -