Fixing the row_size in matmul_lib.py
Hi @cheshire Can you please review this PR? Thank you! @cheshire,
Thank you for the update. I will try to raise the PR for the changes in openxla repo with the test case. 
Fixing the row_size in matmul_lib.py