Update reference submission to handle all batch sizes

[code-perspective]

The commands that needs to work is ./harness/run_submission 1