update rte3 runner with KL_divergence