Add #ifdef GPU_CODEGEN guard to changes of NodeBuilder