MCPcopy
hub / github.com/deepspeedai/DeepSpeed / test_mpi_backend.py

File test_mpi_backend.py

tests/onebit/test_mpi_backend.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1# Copyright (c) Microsoft Corporation.
2# SPDX-License-Identifier: Apache-2.0
3
4# DeepSpeed Team

Callers

nothing calls this directly

Calls 9

get_acceleratorFunction · 0.90
MpiBackendClass · 0.90
torch_simFunction · 0.70
device_countMethod · 0.45
deviceMethod · 0.45
device_nameMethod · 0.45
empty_cacheMethod · 0.45
compressed_allreduceMethod · 0.45

Tested by

no test coverage detected