use non_reentrant_checkpoint fix requires_grad of input must be true for activation checkpoint layer in pipeline train.

use non_reentrant_checkpoint fix requires_grad of input must be true for activation checkpoint layer in pipeline train. #4174