DeepSpeed
a655d7d3 - Switch to torch.linalg.norm (#3984)

Commit

1 year ago

Switch to torch.linalg.norm (#3984) * Switch to torch.linalg.norm * Fix formatting

References

#3984 - Switch to torch.linalg.norm

Author

loadams

loadams

Parents

Files5

deepspeed
- compression
  - basic_layer.py
- runtime
  - comm
    - mpi.py
    - nccl.py
  - fp16/onebit
    - lamb.py
tests/unit/hybrid_engine
- test_he_lora.py

Loading