Frank Lee 638a07a7f9 [test] fixed gemini plugin test (#3411) ...

* [test] fixed gemini plugin test

* polish code

* polish code

2023-04-03 17:12:22 +08:00

..

__init__.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

amp_optimizer.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

base_offload_module.py

[test] fixed gemini plugin test (#3411)

2023-04-03 17:12:22 +08:00

mem_optimize.py

[test] fixed gemini plugin test (#3411)

2023-04-03 17:12:22 +08:00

region_manager.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

region.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

runtime.py

[test] fixed gemini plugin test (#3411)

2023-04-03 17:12:22 +08:00

solver.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

training_simulator.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

util.py

[test] fixed gemini plugin test (#3411)

2023-04-03 17:12:22 +08:00