| .. |
|
ascend
|
!13507 dynamic shape adapter
|
2021-03-20 13:59:06 +08:00 |
|
cpu
|
add cpu micprecision
|
2021-03-19 17:43:08 +08:00 |
|
executor
|
fix issue#I3ARG6
|
2021-03-17 11:53:32 +08:00 |
|
gpu
|
!13344 add DeviceContext module
|
2021-03-19 18:00:36 +08:00 |
|
CMakeLists.txt
|
!13012 add mul fusion based on allreduce fusion
|
2021-03-16 14:04:51 +08:00 |
|
bucket.cc
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
bucket.h
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
convert_tensor_utils.cc
|
fix cpu type excption
|
2020-10-23 15:42:11 +08:00 |
|
convert_tensor_utils.h
|
fix cpu type excption
|
2020-10-23 15:42:11 +08:00 |
|
device_address.h
|
add hardware abstract layer
|
2021-03-19 11:28:53 +08:00 |
|
kernel_adjust.cc
|
enable loop sink when no getnext in execution orders
|
2020-12-29 10:19:10 +08:00 |
|
kernel_adjust.h
|
enable loop sink when no getnext in execution orders
|
2020-12-29 10:19:10 +08:00 |
|
kernel_info.cc
|
refine gpu memory swap performance
|
2020-07-20 20:32:08 +08:00 |
|
kernel_info.h
|
change feature map flag
|
2020-11-27 15:04:52 +08:00 |
|
kernel_runtime.cc
|
workspace of comm op can be reused
|
2021-03-16 14:42:52 +08:00 |
|
kernel_runtime.h
|
Support ms_function + heterogenous
|
2021-03-09 18:59:03 +08:00 |
|
kernel_runtime_manager.cc
|
PyNative AllReduce Bucket
|
2021-03-03 15:47:18 +08:00 |
|
kernel_runtime_manager.h
|
PyNative AllReduce Bucket
|
2021-03-03 15:47:18 +08:00 |
|
launch_kernel.cc
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
launch_kernel.h
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
launch_mul.cc
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
launch_mul.h
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
memory_manager.cc
|
refactor RDR to support single name
|
2021-03-17 11:34:05 +08:00 |
|
memory_manager.h
|
PyNative AllReduce Bucket
|
2021-03-03 15:47:18 +08:00 |