From 414b91bbb3fe8c2a6966ea4a159baf248deecd55 Mon Sep 17 00:00:00 2001
From: Ezra-H <44772185+Ezra-H@users.noreply.github.com>
Date: Sun, 1 Nov 2020 15:21:10 +0800
Subject: [PATCH] Update compressor.py

---
 autodist/kernel/synchronization/compressor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/autodist/kernel/synchronization/compressor.py b/autodist/kernel/synchronization/compressor.py
index 868b542..917f553 100644
--- a/autodist/kernel/synchronization/compressor.py
+++ b/autodist/kernel/synchronization/compressor.py
@@ -300,5 +300,5 @@ def _modified_gram_schmidt(matrix):
 
             rest = matrix[:, (i + 1):]
             rest -= math_ops.reduce_sum_v1(v * rest, axis=0, keepdims=True) * v
-            matrix = array_ops.concat([matrix[:, :i], v, rest], axis=1)
+            matrix = array_ops.concat([matrix[:, :i], v, rest], 1)
         return matrix