unit test

Pass `context_fn` to `activation_checkpoint_params` #2493

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
unit_tests_stable_pytorch (3.8)	5m 18s
unit_tests_nightly_pytorch (3.8)	5m 46s
unit_tests_nightly_pytorch (3.9)	5m 59s
	17m 3s