Unverified Commit 82e267b9 authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

change the condition when execute self.__matmul_nosharding() (#49)


Co-authored-by: default avatarGuoshuai Zhao <guzhao@microsoft.com>
parent 1f726091
......@@ -221,7 +221,7 @@ def _benchmark(self):
K = self._args.k
N = self._args.n
for mode in self._args.mode:
if mode == ShardingMode.NOSHARDING or self.__world_size == 1:
if mode == ShardingMode.NOSHARDING:
elapse_times = self.__matmul_nosharding(M, K, N)
elif mode == ShardingMode.ALLREDUCE:
elapse_times = self.__matmul_allreduce(M, K, N)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment