Adds where to set the compute_driver info.

Change-Id: Ifb47d480d6633dedd5c281eb49ac21caca3ccf8f