DeepSpeed
32b3cbdd
- Merge branch 'master' into cholmes/isolate-src-code
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
1 year ago
Merge branch 'master' into cholmes/isolate-src-code
References
#4610 - Isolate src code and testing for DeepSpeed-FastGen
Author
cmikeh2
Parents
c1b1c2a2
38b41dff
Files
49
.github/workflows
nv-inference.yml
op_builder
inference_cutlass_builder.py
ragged_ops.py
tests/unit/inference
inference_test_utils.py
kernels
__init__.py
core_ops
__init__.py
test_bias_activation.py
test_blas_linear.py
test_gated_activation.py
test_post_ln.py
test_pre_ln.py
test_rms_norm.py
cutlass_ops
__init__.py
test_moe_gemm.py
ragged_ops
__init__.py
ragged_testing_utils.py
test_atom_builder.py
test_blocked_flash.py
test_blocked_kv_copy.py
test_blocked_rotary_emb.py
test_logits_gather.py
test_moe_gather.py
test_moe_scatter.py
test_ragged_embed.py
test_top_1_gating.py
model_implementations
__init__.py
parameters
__init__.py
test_layer_inheritance.py
test_mapping.py
test_multi_parameter_layer.py
test_parameter_list.py
utils.py
sharding
__init__.py
test_attn_out_sharding.py
test_mlp_sharding.py
test_qkv_sharding.py
modules
__init__.py
test_blas_linear_module.py
test_blocked_attn.py
test_cuda_pre_ln_module.py
test_custom_module.py
test_cutlass_moe.py
test_post_ln_module.py
test_pre_rms_module.py
ragged
__init__.py
test_blocked_allocator.py
test_manager_configs.py
test_ragged_wrapper.py
version.txt
Loading