Add mpi in update-codegen.sh (#1600)
Add mpi in update-codegen.sh (#1600)

Signed-off-by: Peng Gao <096299f9e0a0b80c29d84a996900ded94c5ca3f6@gmail.com>
kubeflow/training-operator,kubeflow/training-operator,kubeflow/training-operator