I have a main function calls an inner function which works OK in series manner, i.e. without spmd. However after converting to spmd, an error of 'inner matrix dimensions must agree' happens immediately at the second line of the inner function. I tried with 1 labor, as if the code still runs in series manner, the error also occurs.
So what does it mean by inner matrix? I have some other codes with spmd, which all work fine.
Any one can point me a direction? Many thanks in advance.