matmul bug fix for non blocking