From 6339f62f56a11f1b89cc6ee35d4b6382f698f14b Mon Sep 17 00:00:00 2001
From: Exusial <bilibili39@163.com>
Date: Sat, 27 Aug 2022 22:25:00 +0800
Subject: [PATCH] fix einsum space interpret error.

---
 python/jittor/linalg.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/jittor/linalg.py b/python/jittor/linalg.py
index aeee373c..320efcb2 100644
--- a/python/jittor/linalg.py
+++ b/python/jittor/linalg.py
@@ -486,7 +486,7 @@ def einsum(string, *args):
     
     def einsum_outshape(einsum_expr, inputs):
         shps = np_cpu.concatenate([in_.shape for in_ in inputs])
-        p = einsum_expr.split(',')
+        p = einsum_expr.replace(" ", "").split(',')
         s = p[:-1] + p[-1].split('->')
         if s[-1]=='':
             return ()