mirror of https://github.com/hpcaitech/ColossalAI
dd2bf02679
* 1.support sam 2.add fused qkv for nn.Linear * update utils support set element in list * overtwrite SamVisionAttention foward to use DropoutForParallelInput * remove unused code |
||
|---|---|---|
| .. | ||
| test_dist_crossentropy.py | ||
| test_dropout.py | ||
| test_embedding.py | ||
| test_gpt2_qkv_fused_linear_1d.py | ||
| test_layernorm.py | ||
| test_linear_1d.py | ||
| test_qkv_fused_linear_1d.py | ||
| test_vocab_parallel_embedding_1d.py | ||