unit test

Pass `context_fn` to `activation_checkpoint_params` #2495

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
unit_tests_stable_pytorch (3.8)	5m 24s
unit_tests_nightly_pytorch (3.8)	5m 52s
unit_tests_nightly_pytorch (3.9)	6m 8s
	17m 24s