DeepSpeed

d8ed3ce4 - remove the print line in _einsum_flops_compute function (#1885)