Add support for All-in-one, controller, or compute