Initialize SPMD backend during graph precompile (#4227)