370d01765c
ShapeInfo for the store operand may be dropped, e.g. because the operand got folded by transpose optimizations to another instruction w/o shape info. This was exposed by the assertion added in https://github.com/llvm/llvm-project/pull/142416. This updates VisitStore to use the shape-info directly from the instruction, which is in line with the other Visit* functions and ensures that we won't lose shape info. PR: https://github.com/llvm/llvm-project/pull/142664