xla

51f8d3e6 - Modularize zero step function and make it customizable (#7233)