Fixes multigpu with-context
Further improve multi gpu API