Fix Incorrect output for A or B with dim=1 in GEMM

Fix Incorrect output for A or B with dim=1 in GEMM #5539