Lev Kurilenko 87eaf8f99a Check for local CUDA graphs when enable_cuda_graph=True (#2941) 1 年之前
..
__init__.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
clip_encoder.py 87eaf8f99a Check for local CUDA graphs when enable_cuda_graph=True (#2941) 1 年之前
ds_base.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
ds_bert.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前
ds_bloom.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前
ds_gpt.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前
ds_megatron_gpt.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前
ds_opt.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前
ds_transformer.py 0a73e6e613 Container param cleanup + remove qkv_merging (#2780) 1 年之前