Xuanlei Zhao 30dd13c450 [autochunk] support complete benchmark (#3121) ... * refact memory code * dont log free var memory * add memory align * update chunk target * update setting for new memory * finish test * update tracer * update typo * update test * add unet test * add bench * update bench * update bench * init * support vit * move to cpu * add cpu benchmark		2023-03-13 17:42:37 +08:00
..
benchmark_autochunk_alphafold.py	[autochunk] support complete benchmark (#3121)	2023-03-13 17:42:37 +08:00
test_autochunk_alphafold_utils.py	…
test_autochunk_evoformer_block.py	…
test_autochunk_evoformer_stack.py	…
test_autochunk_extramsa_block.py	…