xla

[benchmarks] Small fixes for benchmarking script.

#6632

Merged