Small cleanups to enable running CePO on GPU systems.#1

bvanessen wants to merge 2 commits intoLBANN:cepoLBANN/cb_optillm:cepofrom

bvanessen:llnl_cepobvanessen/cb_optillm:llnl_cepoCopy head branch name to clipboard