DeepSpeed
3e856464 - Add HE support for the rest of model containers (#3191)

Commit

2 years ago

Add HE support for the rest of model containers (#3191) Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

References

#3191 - Add HE support for the rest of model containers

Author

RezaYazdaniAminabadi

RezaYazdaniAminabadi

Parents

Files10

deepspeed
- module_inject
  - containers
    - base.py
    - bloom.py
    - features
      - megatron.py
    - gptj.py
    - gptneo.py
    - gptneox.py
  - replace_module.py
- ops/transformer/inference
  - ds_attention.py
  - op_binding
    - gelu_gemm.py
- runtime
  - hybrid_engine.py

Loading