|
|
|
@ -12,16 +12,6 @@
|
|
|
|
|
# See the License for the specific language governing permissions and
|
|
|
|
|
# limitations under the License.
|
|
|
|
|
|
|
|
|
|
# For op in FP16_CHECK_OP_LIST, the op test of fp16 precision should inherit OpTestFp16
|
|
|
|
|
FP16_CHECK_OP_LIST = [
|
|
|
|
|
'abs', 'acos', 'asin', 'atan', 'brelu', 'concat', 'cos', 'elementwise_div',
|
|
|
|
|
'elementwise_mul', 'elu', 'exp', 'gelu', 'hard_shrink', 'hard_swish', 'log',
|
|
|
|
|
'logsigmoid', 'mean', 'mul', 'pad', 'pool2d', 'pow', 'reciprocal', 'relu',
|
|
|
|
|
'relu6', 'scale', 'sigmoid', 'sin', 'slice', 'soft_relu', 'softmax',
|
|
|
|
|
'softmax_with_cross_entropy', 'softshrink', 'softsign', 'sqrt', 'square',
|
|
|
|
|
'stanh', 'sum', 'swish', 'tanh', 'tanh_shrink', 'thresholded_relu'
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
# For op in NO_FP64_CHECK_GRAD_OP_LIST, the op test requires check_grad with fp64 precision
|
|
|
|
|
NO_FP64_CHECK_GRAD_OP_LIST = [
|
|
|
|
|
'abs', 'acos', 'add_position_encoding', 'affine_grid', 'asin', 'atan',
|
|
|
|
@ -61,6 +51,3 @@ NO_FP64_CHECK_GRAD_OP_LIST = [
|
|
|
|
|
'unsqueeze', 'unsqueeze2', 'unstack', 'var_conv_2d', 'warpctc',
|
|
|
|
|
'yolov3_loss'
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
# For cases in NO_FP64_CHECK_GRAD_CASES, the op test requires check_grad with fp64 precision
|
|
|
|
|
NO_FP64_CHECK_GRAD_CASES = ['TestFSPOp']
|
|
|
|
|