mindspore/tests/st/ops/gpu
mindspore-ci-bot b15504f6a5 !9327 Cast half to int by rounding down
From: @jonwe
Reviewed-by: @robingrosman,@tom__chen
Signed-off-by: @robingrosman
2020-12-03 05:25:01 +08:00
..
test_acos_grad_op.py Add supports to ACosGrad and AsinGrad 2020-10-28 09:23:20 +08:00
test_acos_op.py fix bug that asin/acos not support fp16 on gpu 2020-10-28 09:37:24 +08:00
test_adam_fusion.py Gpu Adam Fusion 2020-06-23 20:40:45 +08:00
test_adam_op.py mod_SoftmaxCrossEntropyWithLogits 2020-09-03 17:33:20 +08:00
test_addn_op.py clean pylint 2020-05-18 16:42:35 +08:00
test_argmax_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_argmaxwithvalue_op.py gpu add argmaxwithvalue 2020-06-14 14:51:06 +08:00
test_asin_grad_op.py Add supports to ACosGrad and AsinGrad 2020-10-28 09:23:20 +08:00
test_asin_op.py fix bug that asin/acos not support fp16 on gpu 2020-10-28 09:37:24 +08:00
test_assign_add_op.py clean pylint 2020-05-18 16:42:35 +08:00
test_assign_op.py modify_bug 2020-09-09 18:20:32 +08:00
test_batch_matmul.py Gpu Adam Fusion 2020-06-23 20:40:45 +08:00
test_batchnorm_fold2_op.py remove _quant_ops.py from __init__.py 2020-06-18 19:26:22 +08:00
test_batchnorm_fold_grad_op.py remove _quant_ops.py from __init__.py 2020-06-18 19:26:22 +08:00
test_batchnorm_fold_op.py fix GPU hete 2020-09-15 20:36:23 +08:00
test_batchnorm_op.py batchnorm2d use_batch_statistics false test case 2020-11-19 16:23:23 -05:00
test_binary_cross_entropy_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_boundingbox_decode_op.py gpu ops code and test case. 2020-07-22 20:15:52 +08:00
test_boundingbox_encode_op.py gpu ops code and test case. 2020-07-22 20:15:52 +08:00
test_broadcast_op.py DinNoNan gpu kernel supports int8/uint8 2020-11-24 11:38:03 -05:00
test_broadcast_to_ops.py gpu broadcast to 2020-07-29 16:24:01 +08:00
test_cast_op.py Cast half to int by rounding down 2020-12-01 11:48:30 -05:00
test_categorical_op.py add categorical 2020-08-19 11:21:15 +08:00
test_check_valid_op.py type support for faster rcnn gpu kernels 2020-08-12 19:27:05 -04:00
test_cholesky_op.py Add Cholesky op at GPU back-end 2020-11-30 20:27:22 -05:00
test_concatv2_op.py type support for faster rcnn gpu kernels 2020-08-12 19:27:05 -04:00
test_conv2d_backprop_filter_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_conv2d_backprop_input_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_conv2d_depthwiseconv2d.py fix weight_init bug 2020-09-17 09:23:34 +08:00
test_conv2d_op.py gpu optimize the max device memory config 2020-06-17 16:29:05 +08:00
test_correction_mul_grad_op.py remove _quant_ops.py from __init__.py 2020-06-18 19:26:22 +08:00
test_correction_mul_op.py remove _quant_ops.py from __init__.py 2020-06-18 19:26:22 +08:00
test_cos_op.py add GPU operator 2020-09-19 12:13:06 +08:00
test_ctcloss_op.py remove CTCLossV2 2020-09-25 16:09:20 +08:00
test_cudnn_inplace_fusion.py cudnn inplace optimizer 2020-09-30 16:18:16 +08:00
test_cumsum_op.py add gpu cumsum 2020-07-23 15:43:04 +08:00
test_dense.py add_channel_to_attr 2020-11-12 20:18:26 +08:00
test_dense_op.py modify_dense 2020-11-06 15:45:43 +08:00
test_determinant_triangle.py Add calculation of triangle matrix determinant op at GPU back-end 2020-10-27 15:20:23 -04:00
test_div_op.py Add GPU div kernel 2020-08-11 10:18:34 +08:00
test_dropout.py modify some bug and add test case for gpu dropout op 2020-08-10 16:55:56 +08:00
test_dynamic_shape_op.py int64 support and typo fix 2020-11-26 20:47:20 -05:00
test_elu_grad_op.py support ops elu and elugrad on gpu 2020-10-21 15:52:15 +08:00
test_elu_op.py support ops elu and elugrad on gpu 2020-10-21 15:52:15 +08:00
test_equal_op.py Equal op dynamic shape 2020-11-27 14:59:53 -05:00
test_equalcount_op.py clean pylint 2020-05-18 16:42:35 +08:00
test_erf_op.py Add dtype float16 that erf and erfc should support 2020-09-17 09:36:23 -04:00
test_erfc_op.py Add dtype float16 that erf and erfc should support 2020-09-17 09:36:23 -04:00
test_error_on_dynamic_shape_input_op.py initial commit for gpu dynamic shape testing op 2020-11-19 11:22:39 -05:00
test_exp_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_expand_dims.py ExpandDims dynamic shape 2020-11-27 06:24:04 -05:00
test_expm1_op.py Add expm1 op at GPU back-end 2020-10-30 10:06:57 -04:00
test_fake_quant_perchannel.py add fake quant test case for gpu 2020-06-23 15:27:33 +08:00
test_fake_quant_perchannel_grad.py add fake quant test case for gpu 2020-06-23 15:27:33 +08:00
test_fake_quant_perlayer.py add fake quant test case for gpu 2020-06-23 15:27:33 +08:00
test_fake_quant_perlayer_grad.py add fake quant test case for gpu 2020-06-23 15:27:33 +08:00
test_flatten_grad_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_flatten_op.py fix all nop node graph execute 2020-06-18 15:56:38 +08:00
test_float_status_op.py refine data copy in multi-graph 2020-05-27 18:47:15 +08:00
test_floordiv_op.py broadcast, slice, scatter_nd ops optimizer. 2020-08-08 10:30:05 +08:00
test_ftrl_op.py mod_SoftmaxCrossEntropyWithLogits 2020-09-03 17:33:20 +08:00
test_gatherV2_op.py updated to test files + DynamicOpCheckFix 2020-12-02 13:44:46 -05:00
test_gather_grad_op.py fix some bug in gather and gathergrad op 2020-11-23 16:24:45 +08:00
test_gather_op.py add gather op on gpu 2020-10-30 15:05:36 +08:00
test_gathernd_op.py type support for faster rcnn gpu kernels 2020-08-12 19:27:05 -04:00
test_gelu_grad_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_gelu_op.py gpu Gelu kernel support fp16 2020-06-18 19:33:27 +08:00
test_gpu_convert_to_dynamic_shape_op.py initial commit for gpu dynamic shape testing op 2020-11-19 11:22:39 -05:00
test_iou_op.py add iou ops. 2020-07-27 11:12:26 +08:00
test_kl_div_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_l2loss_op.py add support to op L2Loss on gpu 2020-11-27 09:34:47 +08:00
test_l2normalize_grad_op.py new add l2normalize_grad gpu ops. 2020-10-27 17:18:45 +08:00
test_l2normalize_op.py add l2normalize gpu kernel. 2020-10-24 15:53:51 +08:00
test_layer_norm_grad_op.py gpu layernorm 2020-07-30 20:11:04 +08:00
test_layer_norm_op.py gpu layernorm 2020-07-30 20:11:04 +08:00
test_lessequal_op.py int16 uint8 bool supported akg ops 2020-08-11 20:35:31 -04:00
test_lin_space.py Initial Commit - GPU LinSpace 2020-11-26 02:33:55 -05:00
test_log1p_op.py Add log1p operator at GPU back-end and move erf and erf to the unary_op list 2020-10-27 16:22:10 -04:00
test_log_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_logical_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_logsoftmax_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_lstm_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_maximum_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_maxpool_gpu_op.py clean pylint 2020-05-18 16:42:35 +08:00
test_maxpool_grad_gpu_op.py conv pooling pad support NHWC 2020-08-20 17:21:53 +08:00
test_maxpool_with_argmax_gpu_op.py add maxpool_with_argmax/grad cuda kernel 2020-07-24 06:17:14 -08:00
test_maxpool_with_argmax_grad_gpu_op.py add maxpool_with_argmax/grad cuda kernel 2020-07-24 06:17:14 -08:00
test_minimum_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_mirror_pad.py Reworked mirrorPad 2020-10-25 20:27:54 -04:00
test_momentum_op.py mod_SoftmaxCrossEntropyWithLogits 2020-09-03 17:33:20 +08:00
test_mul_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_multinomial_op.py fix multinomial 2020-10-25 17:17:07 +08:00
test_neg_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_nms_with_mask_op.py reduce based nms final pass - speed improv 2020-08-28 13:41:09 -04:00
test_nn_matmul.py 1D*ND 2020-10-12 17:21:07 +08:00
test_normal.py Add random normal op at MindSpore front-end 2020-07-20 23:31:30 -04:00
test_one_hot_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_oneslike_op.py add gpu oneslike kernel 2020-07-18 15:27:07 +08:00
test_pack_op.py add supports to op pack on gpu 2020-11-06 14:19:30 +08:00
test_pad.py New User facing Pad Kernel + ST + Allows for channel padding 2020-09-04 15:13:05 -04:00
test_random_categorical_op.py add support to op RandomCategorical 2020-11-11 19:13:13 +08:00
test_random_choice_with_mask.py add fix to GPU-RandomChoiceWithMask (bitonicsort & testcase) 2020-08-12 19:22:04 -04:00
test_realdiv_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_reciprocal_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_reduce_max_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_reduce_mean_op.py broadcast, slice, scatter_nd ops optimizer. 2020-08-08 10:30:05 +08:00
test_reduce_min_op.py add gpu cumsum 2020-07-23 15:43:04 +08:00
test_reduce_sum_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_relu6_grad_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_relu6_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_relu_grad_op.py Fixing some tiny faults about Pylint in my code(ops) 2020-05-25 21:27:45 +08:00
test_relu_op.py Cast/ReLU dynamic shapelsgpu op supports int32 and int64 2020-11-27 07:03:50 -05:00
test_repeat_elements_op.py change repeat_element op to a composite op 2020-11-19 15:42:22 -05:00
test_reshape_op.py added type support for reshape gpu kernel 2020-08-20 13:52:23 -04:00
test_resize_nearest_neighbor_grad_op.py fix ResizeNearestNeighbor & add util.cuh to support atomicadd for half 2020-07-30 12:31:03 -04:00
test_resize_nearest_neighbor_op.py add unit test for resizeNearestNeighbor gpu op 2020-08-07 16:54:33 -04:00
test_rmsprop.py fix GPU hete 2020-09-15 20:36:23 +08:00
test_roi_align_grad_op.py roi end mode 2020-11-04 13:21:10 -05:00
test_roi_align_op.py roi end mode 2020-11-04 13:21:10 -05:00
test_rsqrt_grad_op.py new add sqrt_grad and rsqrt_grad. 2020-09-17 17:13:25 +08:00
test_sampled_softmax_loss_op.py validate SampledSoftmaxLoss args 2020-11-19 12:43:58 -05:00
test_scatter_add_op.py fix and add testcase for dynamic shape scatteradd/update transpose 2020-11-26 17:33:49 -05:00
test_scatter_nd.py add scatternd ut 2020-08-25 17:03:57 -04:00
test_scatter_update_op.py fix and add testcase for dynamic shape scatteradd/update transpose 2020-11-26 17:33:49 -05:00
test_select_op.py Fixing some tiny faults about Pylint in my code(ops) 2020-05-28 09:52:42 +08:00
test_sequence_mask_op.py initial commit 2020-11-30 14:44:22 -05:00
test_sgd_op.py mod_SoftmaxCrossEntropyWithLogits 2020-09-03 17:33:20 +08:00
test_sigmoid_cross_entropy_with_logits_grad_op.py add SigmoidCrossEntropyWithLogitsGrad op 2020-06-17 19:33:27 +08:00
test_sigmoid_cross_entropy_with_logits_op.py add SigmoidCrossEntropyWithLogits op 2020-06-17 17:00:26 +08:00
test_sigmoid_grad_grad_op.py add grad definition for sigmoidGrad operation 2020-11-23 20:08:46 +08:00
test_sigmoid_grad_op.py add Sigmoid and SigmoidGrad operation of GPU 2020-06-18 15:42:58 +08:00
test_sigmoid_op.py add Sigmoid and SigmoidGrad operation of GPU 2020-06-18 15:42:58 +08:00
test_sin_op.py add GPU operator 2020-09-19 12:13:06 +08:00
test_slice.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_slice_grad.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_smoothl1loss_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_softmax_cross_entropy_with_logits_op.py Fixing some tiny faults about Pylint in my code(ops) 2020-05-25 21:27:45 +08:00
test_softmax_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_softplus_grad_op.py new add softplus and softplus grad gpu ops. 2020-10-29 15:13:55 +08:00
test_softplus_op.py new add softplus and softplus grad gpu ops. 2020-10-29 15:13:55 +08:00
test_sparse_apply_ftrl_op.py sparse apply ftrl gpu kernel 2020-10-28 11:56:04 -04:00
test_sparse_apply_proximal_adagrad_op.py add GPU SparseApplyProximalAdagrad 2020-10-28 14:43:08 -04:00
test_sparse_gather_v2_op.py fix dynamic shape and add testcase for SparseGatherV2 GPU 2020-11-27 12:45:29 -05:00
test_sparse_softmax_cross_entropy_with_logits_op.py mod_SoftmaxCrossEntropyWithLogits 2020-09-03 17:33:20 +08:00
test_split.py change Split op to extend PrimitiveWithCheck 2020-11-27 15:33:20 -05:00
test_sqrt_grad_op.py new add sqrt_grad and rsqrt_grad. 2020-09-17 17:13:25 +08:00
test_sqrt_op.py fix pylint check issues 2020-05-25 15:34:01 +08:00
test_square_op.py Update Square Op to support Dynamic Shape 2020-11-26 12:49:15 -05:00
test_squeeze_op.py int16 uint8 bool supported akg ops 2020-08-11 20:35:31 -04:00
test_standard_normal.py Add test cases for uniform ops on GPU 2020-08-26 12:02:27 -04:00
test_stridedslice_grad_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_stridedslice_op.py gpu stridedslice 2020-08-21 16:21:27 +08:00
test_sub_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
test_tanh_op.py remove name arg from gradoperation 2020-08-26 11:16:57 +08:00
test_tensoradd.py reverse fix shape=0 2020-11-09 15:06:37 +08:00
test_tensordot_op.py initial commit 2020-11-30 17:52:07 -05:00
test_tile_op.py int16 uint8 bool supported akg ops 2020-08-11 20:35:31 -04:00
test_topk_op.py gpu support TopK kernel 2020-07-15 19:51:30 +08:00
test_transpose_op.py fix and add testcase for dynamic shape scatteradd/update transpose 2020-11-26 17:33:49 -05:00
test_uniform_candidate_sampler_op.py rename UniformSampler to UniformCandidateSampler 2020-11-02 16:31:44 -05:00
test_uniform_int.py Add test cases for uniform ops on GPU 2020-08-26 12:02:27 -04:00
test_uniform_real.py Add test cases for uniform ops on GPU 2020-08-26 12:02:27 -04:00
test_unique_op.py Fix unique dynamic shape 2020-12-02 07:49:56 -05:00
test_unpack_op.py add supports to op unpack on gpu 2020-11-06 14:23:54 +08:00
test_unsorted_segment_max.py updated to test files + DynamicOpCheckFix 2020-12-02 13:44:46 -05:00
test_unsorted_segment_min.py updated to test files + DynamicOpCheckFix 2020-12-02 13:44:46 -05:00
test_unsorted_segment_sum.py updated to test files + DynamicOpCheckFix 2020-12-02 13:44:46 -05:00
test_zeroslike_op.py change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00