Add a check_correctness flag in the benchmark