mindspore2022/mindspore/ccsrc/backend/kernel_compiler/cpu
mindspore-ci-bot e212d12257 !13347 add cpu mixprecision
From: @huaweib
Reviewed-by: @zhoufeng54
Signed-off-by:
2021-03-20 15:33:13 +08:00
..
mkldnn !12115 IR operators of GPU and CPU are unified as batchnorm 2021-03-19 14:51:26 +08:00
ps remove static worker 2021-03-09 16:10:52 +08:00
quantum n_threads bug 2021-03-11 18:02:13 +08:00
adam_cpu_kernel.cc add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
adam_cpu_kernel.h add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
adam_delta_cpu_kernel.cc use thread pool for some cpu kernel 2020-12-17 17:11:34 +08:00
adam_delta_cpu_kernel.h add AdamOffload optimizer 2020-11-18 16:13:47 +08:00
allgather_cpu_kernel.cc dlopen cpu mpi adapter 2020-08-09 14:29:23 +08:00
allgather_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
apply_adagrad_cpu_kernel.cc Fix: ApplyAdagrad output. 2021-03-02 10:41:16 +08:00
apply_adagrad_cpu_kernel.h Fix: ApplyAdagrad output. 2021-03-02 10:41:16 +08:00
apply_momentum_cpu_kernel.cc decoupling core and context 2020-08-05 08:15:14 +08:00
apply_momentum_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
argmax_cpu_kernel.cc fix cpu/gpu argmax op 2021-03-04 16:42:41 +08:00
argmax_cpu_kernel.h fix cpu/gpu argmax op 2021-03-04 16:42:41 +08:00
argmin_with_value_cpu_kernel.cc add CPU operators: Div/Mod/Minimum/Floor/ArgMinWithValue 2020-12-26 10:05:12 +08:00
argmin_with_value_cpu_kernel.h add CPU operators: Div/Mod/Minimum/Floor/ArgMinWithValue 2020-12-26 10:05:12 +08:00
arithmetic_cpu_kernel.cc add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
arithmetic_cpu_kernel.h add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
arithmetic_self_cpu_kernel.cc add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
arithmetic_self_cpu_kernel.h add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
assign_cpu_kernel.cc fix assign cpu output 2021-03-10 09:21:17 +08:00
assign_cpu_kernel.h Add assign op for cpu 2020-12-05 10:52:07 +08:00
bias_add_cpu_kernel.cc biasadd operator report error when input is not 2d or 4d 2021-03-08 14:33:05 +08:00
bias_add_cpu_kernel.h biasadd operator report error when input is not 2d or 4d 2021-03-08 14:33:05 +08:00
bias_add_grad_cpu_kernel.cc biasadd operator report error when input is not 2d or 4d 2021-03-08 14:33:05 +08:00
bias_add_grad_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
binary_cross_entropy_cpu_kernel.cc add bce cpu ops 2021-01-27 15:48:19 +08:00
binary_cross_entropy_cpu_kernel.h add bce cpu ops 2021-01-27 15:48:19 +08:00
binary_cross_entropy_grad_kernel.cc add bce cpu ops 2021-01-27 15:48:19 +08:00
binary_cross_entropy_grad_kernel.h add bce cpu ops 2021-01-27 15:48:19 +08:00
broadcast_to_cpu_kernel.cc add cpu broadcast_to op 2021-03-20 10:44:01 +08:00
broadcast_to_cpu_kernel.h add cpu broadcast_to op 2021-03-20 10:44:01 +08:00
cache_swap_hashmap_cpu_kernel.cc fix reshape dynamic and emb 2020-11-20 09:15:58 +08:00
cache_swap_hashmap_cpu_kernel.h add dynamic ops 2020-11-16 11:07:40 +08:00
cast_cpu_kernel.cc add cpu micprecision 2021-03-19 17:43:08 +08:00
cast_cpu_kernel.h add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
concat_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
concat_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
cpu_kernel.cc Use ThreadPool in ParallelFor 2021-01-23 18:23:03 +08:00
cpu_kernel.h fix conv2d_grad_inpu infer strides. 2021-03-10 16:14:12 +08:00
cpu_kernel_factory.cc optimize code compile performance 2020-08-18 14:33:24 +08:00
cpu_kernel_factory.h decoupling core and context 2020-08-05 08:15:14 +08:00
ctcloss_cpu_kernel.cc mod_conv2d_pool_add_ctcloss 2021-01-26 10:30:10 +08:00
ctcloss_cpu_kernel.h mod_conv2d_pool_add_ctcloss 2021-01-26 10:30:10 +08:00
cumsum_cpu_kernel.cc Add CumSum Kerner for CPU 2020-12-25 16:30:11 +08:00
cumsum_cpu_kernel.h Add CumSum Kerner for CPU 2020-12-25 16:30:11 +08:00
debug_cpu_kernel.cc enable debugger by default and set correct log message severity 2020-09-10 21:42:00 -04:00
debug_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
dropout_cpu_kernel.cc fix some liness 2020-12-24 15:44:07 +08:00
dropout_cpu_kernel.h add dropout 2020-12-04 11:25:10 +08:00
dropout_grad_kernel.cc static check clean 2020-12-16 15:15:28 +08:00
dropout_grad_kernel.h static check clean 2020-12-16 15:15:28 +08:00
dynamic_assign_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
dynamic_assign_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
eltwise_grad_cpu_kernel.cc add sinh, cosh, asinh, acosh, atanh, atan2, asinhgrad, acoshgrad for cpu 2021-02-25 10:07:21 +08:00
eltwise_grad_cpu_kernel.h add sinh, cosh, asinh, acosh, atanh, atan2, asinhgrad, acoshgrad for cpu 2021-02-25 10:07:21 +08:00
elu_grad_cpu_kernel.cc Add EluGrad for CPU. 2021-01-26 15:54:26 +08:00
elu_grad_cpu_kernel.h Add EluGrad for CPU. 2021-01-26 15:54:26 +08:00
embedding_look_up_comm_grad_cpu_kernel.cc support int64 shape 2020-11-05 21:15:49 +08:00
embedding_look_up_comm_grad_cpu_kernel.h support int64 shape 2020-11-05 21:15:49 +08:00
embedding_look_up_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
embedding_look_up_cpu_kernel.h !13305 fix pynative memory leak 2021-03-15 20:46:21 +08:00
equal_count_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
equal_count_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
expm1_cpu_kernel.cc expm1 operator 2020-12-15 20:28:07 +08:00
expm1_cpu_kernel.h expm1 operator 2020-12-15 20:28:07 +08:00
gather_cpu_kernel.cc support int64 shape 2020-11-05 21:15:49 +08:00
gather_cpu_kernel.h Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
gather_d_cpu_kernel.cc add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
gather_d_cpu_kernel.h add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
gather_d_grad_cpu_kernel.cc add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
gather_d_grad_cpu_kernel.h add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
gathernd_cpu_kernel.cc add CPU ops: Greater/GreaterEqual/Range/GatherNd for center net 2020-12-29 20:07:54 +08:00
gathernd_cpu_kernel.h add CPU ops: Greater/GreaterEqual/Range/GatherNd for center net 2020-12-29 20:07:54 +08:00
hsigmoid_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hsigmoid_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hsigmoid_grad_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hsigmoid_grad_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hswish_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hswish_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hswish_grad_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
hswish_grad_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
isfinite_cpu_kernel.cc add operator isfinite 2020-12-04 17:25:49 +08:00
isfinite_cpu_kernel.h add operator isfinite 2020-12-04 17:25:49 +08:00
layer_norm_cpu_kernel.cc add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
layer_norm_cpu_kernel.h add some CPU operators 2021-01-15 17:07:39 +08:00
layer_norm_grad_cpu_kernel.cc add parallel for some CPU ops 2021-03-10 15:36:46 +08:00
layer_norm_grad_cpu_kernel.h add some CPU operators 2021-01-15 17:07:39 +08:00
map_cache_idx_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
map_cache_idx_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
map_uniform_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
map_uniform_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
maximum_cpu_kernel.cc add timedistributed and bugfix cpu ops assign and maximum 2020-12-29 17:40:21 +08:00
maximum_cpu_kernel.h Add MaximumGrad 2020-12-09 00:08:00 +08:00
maximum_grad_cpu_kernel.cc add cpu micprecision 2021-03-19 17:43:08 +08:00
maximum_grad_cpu_kernel.h Add MaximumGrad 2020-12-09 00:08:00 +08:00
minimum_cpu_kernel.cc add CPU operators: Div/Mod/Minimum/Floor/ArgMinWithValue 2020-12-26 10:05:12 +08:00
minimum_cpu_kernel.h add CPU operators: Div/Mod/Minimum/Floor/ArgMinWithValue 2020-12-26 10:05:12 +08:00
minimum_grad_cpu_kernel.cc add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
minimum_grad_cpu_kernel.h add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad 2020-12-30 08:10:18 +08:00
mirror_pad_cpu_kernel.cc cpu_pad_op 2021-03-08 16:18:45 +08:00
mirror_pad_cpu_kernel.h cpu_pad_op 2021-03-08 16:18:45 +08:00
mirror_pad_grad_cpu_kernel.cc cpu_pad_op 2021-03-08 16:18:45 +08:00
mirror_pad_grad_cpu_kernel.h cpu_pad_op 2021-03-08 16:18:45 +08:00
one_hot_cpu_kernel.cc support int64 shape 2020-11-05 21:15:49 +08:00
one_hot_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
pack_cpu_kernel.cc add pack for cpu kernel; bugfix adagrad gpu 2020-12-26 16:29:13 +08:00
pack_cpu_kernel.h modify pack to stack 2021-02-04 18:54:31 +08:00
pad_and_shift_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
pad_and_shift_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
pad_cpu_kernel.cc cpu_pad_op 2021-03-08 16:18:45 +08:00
pad_cpu_kernel.h cpu_pad_op 2021-03-08 16:18:45 +08:00
random_cpu_kernel.cc Add SquareDifference kernel for GPU 2020-12-28 16:47:52 +08:00
random_cpu_kernel.h CPU operation dev:StandardNormal 2020-12-04 17:43:03 +08:00
range_cpu_kernel.cc add CPU ops: Greater/GreaterEqual/Range/GatherNd for center net 2020-12-29 20:07:54 +08:00
range_cpu_kernel.h add CPU ops: Greater/GreaterEqual/Range/GatherNd for center net 2020-12-29 20:07:54 +08:00
reduce_cpu_kernel.cc add type support for reduce operations 2021-01-22 14:44:22 +08:00
reduce_cpu_kernel.h add type support for reduce operations 2021-01-22 14:44:22 +08:00
reduce_logic_cpu_kernel.cc add reduceany and reduceall to cpu 2021-02-05 14:57:14 +08:00
reduce_logic_cpu_kernel.h add reduceany and reduceall to cpu 2021-02-05 14:57:14 +08:00
reduce_scatter_cpu_kernel.cc dlopen cpu mpi adapter 2020-08-09 14:29:23 +08:00
reduce_scatter_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
reshape_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
reshape_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
resize_bilinear_cpu_kernel.cc add_resize_ops 2020-12-19 01:06:27 +08:00
resize_bilinear_cpu_kernel.h add_resize_ops 2020-12-19 01:06:27 +08:00
resize_bilinear_grad_cpu_kernel.cc the output of resize_nearest/bilinear ops memset to zero. 2021-01-13 20:44:18 +08:00
resize_bilinear_grad_cpu_kernel.h add_resize_ops 2020-12-19 01:06:27 +08:00
resize_nearest_neighbor_cpu_kernel.cc add_int64_float64 2021-02-26 19:36:24 +08:00
resize_nearest_neighbor_cpu_kernel.h add_int64_float64 2021-02-26 19:36:24 +08:00
resize_nearest_neighbor_grad_cpu_kernel.cc add_int64_float64 2021-02-26 19:36:24 +08:00
resize_nearest_neighbor_grad_cpu_kernel.h add_int64_float64 2021-02-26 19:36:24 +08:00
rmsprop_cpu_kernel.cc [add] support RMSProp CPU Kernel 2020-12-07 09:33:28 +08:00
rmsprop_cpu_kernel.h [add] support RMSProp CPU Kernel 2020-12-07 09:33:28 +08:00
scatter_nd_update_cpu_kernel.cc use thread pool for some cpu kernel 2020-12-17 17:11:34 +08:00
scatter_nd_update_cpu_kernel.h optimize ScatterNdUpdate cpu kernel 2020-10-09 16:01:35 +08:00
search_cache_idx_cpu_kernel.cc fix reshape dynamic and emb 2020-11-20 09:15:58 +08:00
search_cache_idx_cpu_kernel.h add dynamic ops 2020-11-16 11:07:40 +08:00
select_cpu_kernel.cc Add SquareDifference kernel for GPU 2020-12-28 16:47:52 +08:00
select_cpu_kernel.h Add Select ops for cpu 2020-12-04 10:22:33 +08:00
sigmoid_cross_entropy_with_logits_cpu_kernel.cc mac build 2020-12-08 19:13:47 +08:00
sigmoid_cross_entropy_with_logits_cpu_kernel.h add_cpu_op 2020-10-30 13:07:35 +08:00
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.cc mac build 2020-12-08 19:13:47 +08:00
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.h add_cpu_op 2020-10-30 13:07:35 +08:00
slice_cpu_kernel.cc support bool data type for StridedSlice op 2021-03-15 16:02:53 +08:00
slice_cpu_kernel.h support bool data type for StridedSlice op 2021-03-15 16:02:53 +08:00
slice_grad_cpu_kernel.cc support bool data type for StridedSlice op 2021-03-15 16:02:53 +08:00
slice_grad_cpu_kernel.h support bool data type for StridedSlice op 2021-03-15 16:02:53 +08:00
smooth_l1_loss_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
smooth_l1_loss_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
smooth_l1_loss_grad_cpu_kernel.cc Refactor some cpu ops 2021-03-16 20:44:55 +08:00
smooth_l1_loss_grad_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
sparse_apply_adam_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_ftrl_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_ftrl_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_lazy_adam_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_lazy_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_proximal_adagrad_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_proximal_adagrad_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_optimizer_cpu_kernel.h use thread pool for some cpu kernel 2020-12-17 17:11:34 +08:00
split_cpu_kernel.cc Modify split Kernel for CPU 2020-12-31 11:01:46 +08:00
split_cpu_kernel.h Modify split Kernel for CPU 2020-12-31 11:01:46 +08:00
sub_and_filter_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
sub_and_filter_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
tile_cpu_kernel.cc add type support for concat, reshape, transpose, tile, squeeze, expandims cpu op 2021-01-06 13:11:46 +08:00
tile_cpu_kernel.h add type support for concat, reshape, transpose, tile, squeeze, expandims cpu op 2021-01-06 13:11:46 +08:00
topk_cpu_kernel.cc add CPU TopK 2020-12-28 17:34:52 +08:00
topk_cpu_kernel.h add CPU TopK 2020-12-28 17:34:52 +08:00
transpose_cpu_kernel.cc add type support for concat, reshape, transpose, tile, squeeze, expandims cpu op 2021-01-06 13:11:46 +08:00
transpose_cpu_kernel.h add type support for concat, reshape, transpose, tile, squeeze, expandims cpu op 2021-01-06 13:11:46 +08:00
unique_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
unique_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
unique_with_pad_cpu_kernel.cc add unique dynamic 2020-11-14 14:21:43 +08:00
unique_with_pad_cpu_kernel.h optimize cpu unique 2020-11-09 17:05:42 +08:00
unpack_cpu_kernel.cc fix some liness 2020-12-24 15:44:07 +08:00
unpack_cpu_kernel.h modify pack to stack 2021-02-04 18:54:31 +08:00
unsorted_segment_sum_cpu_kernel.cc fix cpu UnsortedSegmentSum 2021-03-16 20:04:45 +08:00
unsorted_segment_sum_cpu_kernel.h add CPU unsortedsegmentsum 2020-12-17 17:10:12 +08:00
update_cache_cpu_kernel.cc fix pynative memory leak 2021-03-15 14:20:50 +08:00
update_cache_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00