Added --device flag to run reference Upstream-commit: bb8ea1fdc1c5c3e1b90c15b451ab22334c86e70c Component: engine