You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Super Daniel 1e88811c7a [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) ... * [autoparallel] first move. * [autoparallel] add solver rotor. * [autoparallel] add ckpt solvers. * [autoparallel] modify codegen. * [fx] fix annotation in test. * [fx] remove check. * [autoparallel] polish docstring. * [fx] refactor MetaTensor.		2 years ago
..
__init__.py	[fx] added activation checkpoint codegen support for torch < 1.12 (#1359)	2 years ago
activation_checkpoint_codegen.py	[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)	2 years ago