mindspore2022/mindspore/ccsrc/runtime/device
mindspore-ci-bot e212d12257 !13347 add cpu mixprecision
From: @huaweib
Reviewed-by: @zhoufeng54
Signed-off-by:
2021-03-20 15:33:13 +08:00
..
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