DeepSpeed
32b3cbdd - Merge branch 'master' into cholmes/isolate-src-code

Comment changes are shownComment changes are hidden
Commit
1 year ago
Merge branch 'master' into cholmes/isolate-src-code
Author
  • .github/workflows
    • File
      nv-inference.yml
  • op_builder
    • File
      inference_cutlass_builder.py
    • File
      ragged_ops.py
  • tests/unit/inference
    • File
      inference_test_utils.py
    • kernels
      • File
        __init__.py
      • core_ops
        • File
          __init__.py
        • File
          test_bias_activation.py
        • File
          test_blas_linear.py
        • File
          test_gated_activation.py
        • File
          test_post_ln.py
        • File
          test_pre_ln.py
        • File
          test_rms_norm.py
      • cutlass_ops
        • File
          __init__.py
        • File
          test_moe_gemm.py
      • ragged_ops
        • File
          __init__.py
        • File
          ragged_testing_utils.py
        • File
          test_atom_builder.py
        • File
          test_blocked_flash.py
        • File
          test_blocked_kv_copy.py
        • File
          test_blocked_rotary_emb.py
        • File
          test_logits_gather.py
        • File
          test_moe_gather.py
        • File
          test_moe_scatter.py
        • File
          test_ragged_embed.py
        • File
          test_top_1_gating.py
    • model_implementations
      • File
        __init__.py
      • parameters
        • File
          __init__.py
        • File
          test_layer_inheritance.py
        • File
          test_mapping.py
        • File
          test_multi_parameter_layer.py
        • File
          test_parameter_list.py
        • File
          utils.py
      • sharding
        • File
          __init__.py
        • File
          test_attn_out_sharding.py
        • File
          test_mlp_sharding.py
        • File
          test_qkv_sharding.py
    • modules
      • File
        __init__.py
      • File
        test_blas_linear_module.py
      • File
        test_blocked_attn.py
      • File
        test_cuda_pre_ln_module.py
      • File
        test_custom_module.py
      • File
        test_cutlass_moe.py
      • File
        test_post_ln_module.py
      • File
        test_pre_rms_module.py
    • ragged
      • File
        __init__.py
      • File
        test_blocked_allocator.py
      • File
        test_manager_configs.py
      • File
        test_ragged_wrapper.py
  • File
    version.txt