Add code to forked version to run on CPU/ACC