| .. |
|
allreduce_fusion
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
auto_parallel
|
IR operators of GPU and CPU are unified as batchnorm
|
2021-03-18 19:02:28 +08:00 |
|
cache_embedding
|
remove embeddinglookup input0 int64, and fix cache
|
2021-03-13 16:47:01 +08:00 |
|
graph_util
|
Do not get shape for monad type
|
2021-02-22 21:35:34 +08:00 |
|
ops_info
|
[bugfix] not find MakeTuple in functional.py
|
2021-03-13 17:45:42 +08:00 |
|
pipeline_transformer
|
Eliminate all useless nodes related to UpdateStates.
|
2021-02-26 14:32:50 +08:00 |
|
strategy_checkpoint
|
add group ckpt
|
2021-01-21 14:41:05 +08:00 |
|
tensor_layout
|
fix auto_parallel_log_bug
|
2020-12-17 16:17:22 +08:00 |
|
CMakeLists.txt
|
add compile option H to visible hidden interface
|
2021-03-04 22:52:48 +08:00 |
|
context.cc
|
!12900 Add communication parallel mode.
|
2021-03-06 15:55:49 +08:00 |
|
context.h
|
Add communication parallel mode.
|
2021-03-05 21:36:03 +08:00 |
|
costmodel_context.cc
|
support for-loop in auto-parallel
|
2020-11-09 17:18:30 +08:00 |
|
costmodel_context.h
|
support for-loop in auto-parallel
|
2020-11-09 17:18:30 +08:00 |
|
device.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
device_manager.cc
|
update pipeline parallel
|
2020-11-18 13:55:34 +08:00 |
|
device_manager.h
|
add group ckpt
|
2021-01-21 14:41:05 +08:00 |
|
device_matrix.cc
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
device_matrix.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
dynamic_creator.h
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
group_manager.cc
|
add group ckpt
|
2021-01-21 14:41:05 +08:00 |
|
group_manager.h
|
add group ckpt
|
2021-01-21 14:41:05 +08:00 |
|
node_check.cc
|
modify pack to stack
|
2021-02-04 18:54:31 +08:00 |
|
node_check.h
|
move parallel-related black-list to core/ir, and fix the cloneCNode bug
|
2020-12-10 15:57:01 +08:00 |
|
status.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
step_auto_parallel.cc
|
[auto-monad] Support side-effects by auto-monad
|
2021-02-08 09:01:15 +08:00 |
|
step_auto_parallel.h
|
simplifying step-auto-parallel
|
2020-12-07 18:36:32 +08:00 |
|
step_parallel.cc
|
fix grad accu bug for no used parameter
|
2021-03-16 10:06:32 +08:00 |
|
step_parallel.h
|
mini step grad accumulation
|
2021-03-01 10:20:12 +08:00 |
|
strategy.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
dingpeifei
87e41aaeee