unit test

Pass `context_fn` to `activation_checkpoint_params` #2494

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
unit_tests_nightly_pytorch (3.8)	5m 56s
unit_tests_stable_pytorch (3.8)	5m 23s
unit_tests_nightly_pytorch (3.9)	6m 13s
	17m 32s