spmd_compile: cleanup