fix setup memory bug on ALCF
fix setup memory bug on ALCF

specter119/custodian,xhqu1981/custodian,davidwaroquiers/custodian,materialsproject/custodian,materialsproject/custodian,materialsproject/custodian,specter119/custodian,specter119/custodian