mindstudio-probe 1.0.1__py3-none-any.whl

msprobe/pytorch/compare/mapping.yaml

@@ -0,0 +1,607 @@
+__and__: __and__
+__iand__: __iand__
+__ilshift__: __ilshift__
+__ior__: __ior__
+__irshift__: __irshift__
+__ixor__: __ixor__
+__lshift__: __lshift__
+__or__: __or__
+__rshift__: __rshift__
+__xor__: __xor__
+_adaptive_avg_pool2d: adaptive_avg_pool2d
+_adaptive_avg_pool3d: adaptive_avg_pool3d
+_cdist_forward: cdist
+_cudnn_rnn: rnn
+_embedding_bag: embedding_bag
+_fft_c2c: fft
+_fft_c2r: rfft
+_foreach_add_: _foreach_add_
+_foreach_addcdiv: _foreach_addcdiv
+_foreach_copy_: _foreach_copy_
+_foreach_lerp_: _foreach_lerp_
+_foreach_maximum: _foreach_maximum
+_foreach_mul: _foreach_mul
+_foreach_neg_: _foreach_neg_
+_foreach_pow: _foreach_pow
+_foreach_reciprocal_: _foreach_reciprocal_
+_foreach_sign: _foreach_sign
+_foreach_sqrt: _foreach_sqrt
+_foreach_sqrt_: _foreach_sqrt_
+_foreach_sub: _foreach_sub
+_fused_adam: FusedAdam
+_linalg_det: det
+_linalg_eigh: eigh
+_linalg_slogdet: slogdet
+_linalg_svd: svd
+_list_to_tensor: as_tensor
+_log_softmax: log_softmax
+_native_batch_norm_legit: batch_norm
+_nested_tensor_from_tensor_list: _nested_tensor_from_tensor_list
+_pdist_forward: pdist
+_pin_memory: pin_memory
+_reshape_alias: reshape
+_resize_output_: resize_
+_softmax: softmax
+_to_copy: to
+abs: abs
+abs_: abs_
+absolute: abs
+absolute_: abs_
+acos: acos
+acos_: acos_
+acosh: acosh
+acosh_: acosh_
+adaptive_max_pool2d: adaptive_max_pool2d
+adaptive_max_pool3d: adaptive_max_pool3d
+add: add
+add_: add_
+addbmm: addbmm
+addbmm_: addbmm_
+addcdiv: addcdiv
+addcdiv_: addcdiv_
+addcmul: addcmul
+addcmul_: addcmul_
+addmm: addmm
+addmm_: addmm_
+addmv: addmv
+addmv_: addmv_
+addr: addr
+affine_grid_generator: affine_grid
+alias: alias
+all: all
+alpha_dropout: AlphaDropout
+amax: amax
+amin: amin
+aminmax: aminmax
+angle: angle
+any: any
+arange: arange
+arccos: acos
+arccos_: arccos_
+arccosh: arccosh
+arccosh_: arccosh_
+arcsin: asin
+arcsin_: arcsin_
+arcsinh: asinh
+arcsinh_: arcsinh_
+arctan: atan
+arctan2: atan2
+arctan2_: arctan2_
+arctan_: arctan_
+arctanh: arctanh
+arctanh_: arctanh_
+argmax: argmax
+argmin: argmin
+argsort: argsort
+as_strided: as_strided
+asin: asin
+asin_: asin_
+asinh: asinh
+asinh_: asinh_
+atan: atan
+atan2: atan2
+atan2_: atan2_
+atan_: atan_
+atanh: atanh
+atanh_: atanh_
+avg_pool2d: avg_pool2d
+avg_pool3d: avg_pool3d
+baddbmm: baddbmm
+baddbmm_: baddbmm_
+bernoulli: bernoulli
+bernoulli_: bernoulli_
+binary_cross_entropy: BCELoss
+binary_cross_entropy_with_logits: binary_cross_entropy_with_logits
+bitwise_and: bitwise_and
+bitwise_and_: bitwise_and_
+bitwise_left_shift: __lshift__
+bitwise_left_shift_: bitwise_left_shift_
+bitwise_not: bitwise_not
+bitwise_not_: bitwise_not_
+bitwise_or: bitwise_or
+bitwise_or_: bitwise_or_
+bitwise_right_shift: __rshift__
+bitwise_right_shift_: bitwise_right_shift_
+bitwise_xor: bitwise_xor
+bitwise_xor_: bitwise_xor_
+bmm: bmm
+broadcast_tensors: broadcast_tensors
+bucketize: bucketize
+cat: cat
+cauchy: Cauchy
+cauchy_: cauchy_
+ceil: ceil
+ceil_: ceil_
+celu: celu
+celu_: celu_
+cholesky: cholesky
+cholesky_inverse: cholesky_inverse
+cholesky_solve: cholesky_solve
+clamp: clamp
+clamp_: clamp_
+clamp_max: clamp_max
+clamp_max_: clamp_max_
+clamp_min: clamp_min
+clamp_min_: clamp_min_
+clip: clip
+clip_: clip_
+clone: clone
+col2im: col2im
+complex: complex
+conj_physical: conj
+conj_physical_: conj_
+constant_pad_nd: pad
+convolution: Conv2d
+copy: copy_
+copy_: copy_
+copysign: copysign
+copysign_: copysign_
+cos: cos
+cos_: cos_
+cosh: cosh
+cosh_: cosh_
+count_nonzero: count_nonzero
+cudnn_batch_norm: BatchNorm2d
+cummax: cummax
+cummin: cummin
+cumprod: cumprod
+cumprod_: cumprod_
+cumsum: cumsum
+cumsum_: cumsum_
+deg2rad: deg2rad
+deg2rad_: deg2rad_
+detach: detach
+diag: diag
+diag_embed: diag_embed
+diagonal: diagonal
+diagonal_copy: diagonal
+diagonal_scatter: diagonal
+digamma: digamma
+digamma_: digamma_
+dist: dist
+div: div
+div_: div_
+divide: div
+divide_: divide_
+dot: dot
+dropout: dropout
+elu: ELU
+elu_: elu_
+embedding: embedding
+empty_like: empty_like
+empty_strided: empty_strided
+eq: eq
+eq_: eq_
+erf: erf
+erf_: erf_
+erfc: erfc
+erfc_: erfc_
+erfinv: erfinv
+erfinv_: erfinv_
+exp: exp
+exp2: exp2
+exp2_: exp2_
+exp_: exp_
+expand: expand
+expm1: expm1
+expm1_: expm1_
+exponential: Exponential
+exponential_: exponential_
+eye: eye
+fft_fft: fft
+fft_fft2: fft2
+fft_fftn: fftn
+fft_fftshift: fftshift
+fft_hfft: hfft
+fft_hfft2: hfft2
+fft_hfftn: hfftn
+fft_ifft: ifft
+fft_ifft2: ifft2
+fft_ifftn: ifftn
+fft_ifftshift: ifftshift
+fft_ihfft: ihfft
+fft_ihfft2: ihfft2
+fft_ihfftn: ifftn
+fft_irfft: irfft
+fft_irfft2: irfft2
+fft_irfftn: irfftn
+fft_rfft: rfft
+fft_rfft2: rfft2
+fft_rfftn: rfftn
+fill: fill_
+fill_: fill_
+fix: fix
+fix_: fix_
+flip: flip
+float_power_: float_power_
+floor: floor
+floor_: floor_
+floor_divide: floor_divide
+floor_divide_: floor_divide_
+fmax: fmax
+fmin: fmin
+fmod: fmod
+fmod_: fmod_
+frac: frac
+frac_: frac_
+full: full
+full_like: full_like
+gather: gather
+gcd: gcd
+gcd_: gcd_
+ge: ge
+ge_: ge_
+gelu: GELU
+gelu_: gelu_
+geometric: Geometric
+geometric_: geometric_
+glu: glu
+greater: gt
+greater_: ge_
+greater_equal: ge
+greater_equal_: ge_
+grid_sampler_2d: grid_sample
+grid_sampler_3d: grid_sample
+gru: GRU
+gt: gt
+gt_: gt_
+hardshrink: Hardshrink
+hardsigmoid: hardsigmoid
+hardsigmoid_: hardsigmoid_
+hardswish: hardswish
+hardswish_: hardswish_
+hardtanh: hardtanh
+hardtanh_: hardtanh_
+heaviside: heaviside
+heaviside_: heaviside_
+hinge_embedding_loss: HingeEmbeddingLoss
+huber_loss: huber_loss
+hypot: hypot
+hypot_: hypot_
+i0: i0
+i0_: i0_
+igamma: igamma
+igamma_: igamma_
+igammac: igammac
+igammac_: igammac_
+index: __getitem__
+index_add: index_add
+index_add_: index_add_
+index_copy: index_copy_
+index_copy_: index_copy_
+index_fill: index_fill_
+index_fill_: index_fill_
+index_put: index_put_
+index_put_: index_put_
+index_reduce: index_select
+index_select: index_select
+is_pinned: is_pinned
+is_same_size: is_same_size
+isinf: isinf
+isnan: isnan
+isneginf: isneginf
+isposinf: isposinf
+istft: istft
+item: item
+lcm: lcm
+lcm_: lcm_
+le: le
+le_: le_
+leaky_relu: LeakyReLU
+leaky_relu_: leaky_relu_
+lerp: lerp
+lerp_: lerp_
+less: less
+less_: less_
+less_equal: le
+less_equal_: less_equal_
+lgamma: lgamma
+lgamma_: lgamma_
+linalg_cholesky_ex: cholesky
+linalg_cross: cross
+linalg_householder_product: householder_product
+linalg_inv_ex: inv
+linalg_ldl_factor_ex: ldl
+linalg_ldl_solve: ldl_solve
+linalg_lu: lu
+linalg_lu_factor_ex: lu_factor
+linalg_lu_solve: lu_solve
+linalg_matrix_exp: matrix_exp
+linalg_qr: qr
+linalg_solve_triangular: solve
+linalg_vector_norm: norm
+linspace: linspace
+log: log
+log10: log10
+log10_: log10_
+log1p: log1p
+log1p_: log1p_
+log2: log2
+log2_: log2_
+log_: log_
+log_normal: LogNormal
+log_sigmoid_forward: log_sigmoid
+logaddexp: logaddexp
+logaddexp2: logaddexp2
+_native_batch_norm_legit_functional: batch_norm
+logcumsumexp: logcumsumexp
+logical_and: logical_and
+logical_and_: logical_and_
+logical_not: logical_not
+logical_not_: logical_not_
+logical_or: logical_or
+logical_or_: logical_or_
+logical_xor: logical_xor
+logical_xor_: logical_xor_
+logit: logit
+logit_: logit_
+logspace: logspace
+logsumexp: logsumexp
+lstm: LSTM
+lt: lt
+lt_: lt_
+lu_unpack: lu_unpack
+margin_ranking_loss: margin_ranking_loss
+masked_fill: masked_fill
+masked_fill_: masked_fill_
+matmul: matmul
+max: max
+max_pool2d_with_indices: MaxPool2d
+max_pool3d_with_indices: MaxPool3d
+max_unpool2d: MaxUnpool2d
+max_unpool3d: max_unpool3d
+maximum: maximum
+mean: mean
+median: median
+meshgrid: meshgrid
+min: min
+minimum: minimum
+mish: Mish
+mish_: mish_
+mm: mm
+mode: mode
+mse_loss: mse_loss
+mul: mul
+mul_: mul_
+multi_margin_loss: MultiMarginLoss
+multilabel_margin_loss_forward: multilabel_margin_loss
+multinomial: multinomial
+multiply: multiply
+multiply_: mul_
+mv: mv
+mvlgamma: mvlgamma
+mvlgamma_: mvlgamma_
+name: name
+nan_to_num: nan_to_num
+nan_to_num_: nan_to_num_
+nanmedian: nanmedian
+nansum: nansum
+narrow_copy: narrow
+native_batch_norm: BatchNorm2d
+native_dropout: dropout
+native_group_norm: group_norm
+native_layer_norm: LayerNorm
+ne: ne
+ne_: ne_
+neg: neg
+neg_: neg_
+negative: neg
+negative_: neg_
+new_empty: new_empty
+new_empty_strided: new_empty_strided
+new_full: new_full
+new_ones: new_ones
+new_zeros: new_zeros
+nextafter: nextafter
+nextafter_: nextafter_
+nll_loss: nll_loss
+nll_loss2d_forward: NLLLoss2d
+nll_loss_forward: NLLLoss
+nonzero_static: nonzero
+norm: norm
+normal: normal
+normal_: normal_
+not_equal: ne
+not_equal_: ne_
+ones: ones
+ones_like: ones_like
+ormqr: ormqr
+pairwise_distance: pairwise_distance
+pdist: pdist
+permute: permute
+pin_memory: pin_memory
+pixel_shuffle: PixelShuffle
+polar: polar
+polygamma: polygamma
+positive: positive
+pow: pow
+pow_: pow_
+prelu: prelu
+prod: prod
+quantized_gru: GRU
+quantized_lstm: LSTM
+rad2deg: rad2deg
+rad2deg_: rad2deg_
+rand: rand
+rand_like: rand_like
+randint: randint
+randint_like: randint_like
+randn: randn
+randn_like: randn_like
+randperm: randperm
+reciprocal: reciprocal
+reciprocal_: reciprocal_
+reflection_pad1d: reflection_pad1d
+reflection_pad2d: reflection_pad2d
+reflection_pad3d: ReflectionPad3d
+relu: relu
+relu6: relu6
+relu_: relu_
+remainder: remainder
+remainder_: remainder_
+renorm: renorm
+renorm_: renorm_
+repeat: repeat
+repeat_interleave: repeat_interleave
+replication_pad1d: ReplicationPad1d
+replication_pad2d: replication_pad2d
+replication_pad3d: replication_pad3d
+resize_as_: resize_as_
+rnn_relu: RNN
+rnn_tanh: RNN
+roll: roll
+rot90: rot90
+round: round
+round_: round_
+rrelu_with_noise: RReLU
+rrelu_with_noise_: rrelu_with_noise
+rsqrt: rsqrt
+rsqrt_: rsqrt_
+rsub: rsub
+scalar_tensor: scalar_tensor
+scatter: scatter_
+scatter_: scatter_
+scatter_add: scatter_add
+scatter_add_: scatter_add_
+searchsorted: searchsorted
+select: select
+selu: selu
+selu_: selu_
+sgn: sgn
+sgn_: sgn_
+sigmoid: sigmoid
+sigmoid_: sigmoid_
+sign: sign
+sign_: sign_
+signbit: signbit
+silu: silu
+silu_: silu_
+sin: sin
+sin_: sin_
+sinc: sinc
+sinc_: sinc_
+sinh: sinh
+sinh_: sinh_
+slice: slice
+smooth_l1_loss: smooth_l1_loss
+soft_margin_loss: soft_margin_loss
+softplus: softplus
+softshrink: softshrink
+sort: sort
+special_airy_ai: airy_ai
+special_bessel_j0: j0
+special_bessel_j1: j1
+special_bessel_y0: y0
+special_bessel_y1: y1
+special_chebyshev_polynomial_t: chebyshev_t
+special_chebyshev_polynomial_u: chebyshev_u
+special_entr: entr
+special_erfcx: erfcx
+special_hermite_polynomial_h: hermite
+special_hermite_polynomial_he: he
+special_i0: i0
+special_i0e: i0e
+special_i1: i1
+special_i1e: i1e
+special_laguerre_polynomial_l: laguerre_l
+special_log_ndtr: log_ndtr
+special_modified_bessel_i0: i0
+special_modified_bessel_i1: i1
+special_modified_bessel_k0: k0
+special_modified_bessel_k1: i1
+special_ndtr: ndtr
+special_ndtri: ndtri
+special_scaled_modified_bessel_k0: i0e
+special_scaled_modified_bessel_k1: scaled_modified_bessel_k1
+special_spherical_bessel_j0: spherical_jn
+special_xlog1py: xlog1py
+special_zeta: zeta
+split: split
+split_with_sizes: split
+sqrt: sqrt
+sqrt_: sqrt_
+square: square
+square_: square_
+squeeze: squeeze
+stack: stack
+std: std
+std_mean: std_mean
+stft: stft
+sub: sub
+sub_: sub_
+subtract: sub
+subtract_: subtract_
+sum: sum
+t: t
+t_: t_
+take: take
+tan: tan
+tan_: tan_
+tanh: tanh
+tanh_: tanh_
+threshold: threshold
+threshold_: threshold_
+to: to
+topk: topk
+trace: trace
+transpose: transpose
+transpose_: transpose_
+triangular_solve: triangular_solve
+tril: tril
+tril_: tril_
+tril_indices: tril_indices
+triu: triu
+triu_: triu_
+triu_indices: triu_indices
+true_divide: true_divide
+true_divide_: true_divide_
+trunc: trunc
+trunc_: trunc_
+unbind: unbind
+unfold: unfold
+uniform: Uniform
+uniform_: uniform_
+unsafe_chunk: unsafe_chunk
+unsafe_split: split
+unsafe_split_with_sizes: split_with_sizes
+unsqueeze: unsqueeze
+unsqueeze_: unsqueeze_
+upsample_bicubic2d: interpolate
+upsample_bilinear2d: upsample_bilinear
+upsample_nearest1d: interpolate
+upsample_nearest2d: interpolate
+upsample_nearest3d: interpolate
+var: var
+var_mean: var_mean
+vdot: vdot
+view: view
+where: where
+xlogy: xlogy
+xlogy_: xlogy_
+zero: zeros
+zero_: zero_
+zeros: zeros
+zeros_like: zeros_like
+
+
+

msprobe/pytorch/compare/match.py

@@ -0,0 +1,36 @@
+import os
+import yaml
+from msprobe.core.common.file_check import FileOpen
+from msprobe.core.common.utils import CompareException
+
+
+class AtenIrMapping():
+    def __init__(self):
+        cur_path = os.path.dirname(os.path.realpath(__file__))
+        yaml_path = os.path.join(cur_path, "mapping.yaml")
+        with FileOpen(yaml_path, 'r') as f:
+            self.aten_mapping = yaml.safe_load(f)
+
+    def match(self, op1, op2):
+        if "Aten" in op1 and "Aten" not in op2:
+            return self.match_op(op1, op2)
+        else:
+            return self.match_op(op2, op1)
+
+    def match_op(self, aten_op, torch_op):
+        try:
+            aten_op_raw_name_overload = '_'.join(aten_op.split("_")[1:-3])
+            aten_op_raw_name = aten_op_raw_name_overload.split('.')[0]
+            torch_op_raw_name = '_'.join(torch_op.split("_")[1:-3]).lower()
+        except IndexError as e:
+            err_msg = f"Dump op name format error: {aten_op}, {torch_op}. Your dump data may be corrupted."
+            raise CompareException.INVALID_DATA_ERROR(err_msg) from e
+        matching_op = self.aten_mapping.get(aten_op_raw_name)
+        if matching_op is None:
+            return False
+        if matching_op.lower() == torch_op_raw_name:
+            return True
+        return False
+
+
+graph_mapping = AtenIrMapping()