Fix bugs for row tiling, now on col tiling
Minor changes
Fix bugs to input dimension constraint
Merge pull request #222 from GabrielWen/diag_flat_fix

Bug fix: diagflat