1from mindspore.common.api import set_adapter_config 2from mindspore._extends.parse import trope as T 3from mindspore._extends.parse.resources import convert_object_map 4from ._register.ms_adapter_api import Tensor, Parameter 5from ._register import multitype_ops 6from ._register import standard_method as S 7 8# Update convert_object_map 9convert_object_map[T.add] = multitype_ops.add 10convert_object_map[T.sub] = multitype_ops.sub 11convert_object_map[T.mul] = multitype_ops.mul 12convert_object_map[T.truediv] = multitype_ops.div 13convert_object_map[T.getitem] = multitype_ops.getitem 14convert_object_map[T.setitem] = multitype_ops.setitem 15convert_object_map[T.floordiv] = multitype_ops.floordiv 16convert_object_map[T.mod] = multitype_ops.mod 17convert_object_map[T.pow] = multitype_ops.pow_ 18convert_object_map[T.and_] = multitype_ops.bitwise_and 19convert_object_map[T.or_] = multitype_ops.bitwise_or 20convert_object_map[T.xor] = multitype_ops.bitwise_xor 21convert_object_map[T.neg] = multitype_ops.negative 22convert_object_map[T.not_] = multitype_ops.logical_not 23convert_object_map[T.eq] = multitype_ops.equal 24convert_object_map[T.ne] = multitype_ops.not_equal 25convert_object_map[T.lt] = multitype_ops.less 26convert_object_map[T.gt] = multitype_ops.greater 27convert_object_map[T.le] = multitype_ops.less_equal 28convert_object_map[T.ge] = multitype_ops.greater_equal 29convert_object_map[T.contains] = multitype_ops.in_ 30convert_object_map[T.not_contains] = multitype_ops.not_in_ 31convert_object_map[T.matmul] = S.adapter_matmul 32convert_object_map[T.invert] = S.adapter_invert 33convert_object_map[T.abs] = S.adapter_abs 34convert_object_map[T.round] = S.adapter_round 35convert_object_map[T.max] = S.adapter_max 36convert_object_map[T.min] = S.adapter_min 37convert_object_map[T.sum] = S.adapter_sum 38 39 40# map for adapter tensor convert 41convert_adapter_tensor_map = {} 42convert_adapter_tensor_map["Tensor"] = S.create_adapter_tensor 43 44 45adapter_config = {"Tensor": Tensor, "Parameter": Parameter, "convert_object_map": convert_object_map, 46 "convert_adapter_tensor_map": convert_adapter_tensor_map} 47set_adapter_config(adapter_config) 48 49 50__all__ = ["Tensor", "Parameter"] 51