mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-29 03:04:55 +08:00
[ONNX] Add bloom ops (#94761)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94761 Approved by: https://github.com/justinchuby
This commit is contained in:
AllenTiTaiWang
committed by
PyTorch MergeBot
PyTorch MergeBot
parent
5ed7c701a3
commit
84a5aec8c6
@ -58,22 +58,29 @@ def aten_getitem(self, i):
|
||||
|
||||
# A simple lookup table for atenlib functions
|
||||
_ATENLIB_FUNCTIONS = {
|
||||
"getitem": aten_getitem,
|
||||
"prims::convert_element_type": prims_convert_element_type,
|
||||
"aten::abs": ops.core.aten_abs,
|
||||
"aten::acos": ops.core.aten_acos,
|
||||
"aten::acosh": ops.core.aten_acosh,
|
||||
"aten::adaptive_avg_pool1d": ops.nn.aten_adaptive_avg_pool1d,
|
||||
"aten::adaptive_avg_pool2d": ops.nn.aten_adaptive_avg_pool2d,
|
||||
"aten::adaptive_avg_pool3d": ops.nn.aten_adaptive_avg_pool3d,
|
||||
"aten::add": ops.core.aten_add,
|
||||
"aten::addmm": ops.core.aten_addmm,
|
||||
"aten::alias": ops.core.aten_alias,
|
||||
"aten::amax": ops.core.aten_amax,
|
||||
"aten::amin": ops.core.aten_amin,
|
||||
"aten::arange": ops.core.aten_arange_start,
|
||||
"aten::argmax": ops.core.aten_argmax,
|
||||
"aten::argmin": ops.core.aten_argmin,
|
||||
"aten::asin": ops.core.aten_asin,
|
||||
"aten::asinh": ops.core.aten_asinh,
|
||||
"aten::atan": ops.core.aten_atan,
|
||||
"aten::atanh": ops.core.aten_atanh,
|
||||
"aten::baddbmm": ops.core.aten_baddbmm,
|
||||
"aten::bitwise_not": ops.core.aten_bitwise_not,
|
||||
"aten::bmm": ops.core.aten_bmm,
|
||||
"aten::ceil": ops.core.aten_ceil,
|
||||
"aten::celu": ops.nn.aten_celu,
|
||||
"aten::clamp_max": ops.core.aten_clamp_max,
|
||||
"aten::clamp_min": ops.core.aten_clamp_min,
|
||||
"aten::clamp": ops.core.aten_clamp,
|
||||
@ -81,69 +88,68 @@ _ATENLIB_FUNCTIONS = {
|
||||
"aten::convolution": ops.core.aten_convolution,
|
||||
"aten::cos": ops.core.aten_cos,
|
||||
"aten::cosh": ops.core.aten_cosh,
|
||||
"aten::cumsum": ops.core.aten_cumsum,
|
||||
"aten::detach": ops.core.aten_detach,
|
||||
"aten::div": ops.core.aten_div,
|
||||
"aten::dot": ops.core.aten_dot,
|
||||
"aten::empty": ops.core.aten_empty,
|
||||
"aten::elu": ops.nn.aten_elu,
|
||||
"aten::embedding": ops.core.aten_embedding,
|
||||
"aten::empty_like": ops.core.aten_empty_like,
|
||||
"aten::empty": ops.core.aten_empty,
|
||||
"aten::eq": ops.core.aten_eq,
|
||||
"aten::equal": ops.core.aten_equal,
|
||||
"aten::erf": ops.core.aten_erf,
|
||||
"aten::exp": ops.core.aten_exp,
|
||||
"aten::exp2": ops.core.aten_exp2,
|
||||
"aten::expand": ops.core.aten_expand,
|
||||
"aten::erf": ops.core.aten_erf,
|
||||
"aten::fmod": ops.core.aten_fmod,
|
||||
"aten::full": ops.core.aten_full,
|
||||
"aten::full_like": ops.core.aten_full_like,
|
||||
"aten::full": ops.core.aten_full,
|
||||
"aten::ge": ops.core.aten_ge,
|
||||
"aten::gelu": ops.nn.aten_gelu,
|
||||
"aten::gt": ops.core.aten_gt,
|
||||
"aten::isinf": ops.core.aten_isinf,
|
||||
"aten::log": ops.core.aten_log,
|
||||
"aten::le": ops.core.aten_le,
|
||||
"aten::leaky_relu": ops.nn.aten_leaky_relu,
|
||||
"aten::linear": ops.nn.aten_linear,
|
||||
"aten::log_softmax": ops.special.aten_special_log_softmax,
|
||||
"aten::log": ops.core.aten_log,
|
||||
"aten::log10": ops.core.aten_log10,
|
||||
"aten::log1p": ops.core.aten_log1p,
|
||||
"aten::log_softmax": ops.special.aten_special_log_softmax,
|
||||
"aten::log2": ops.core.aten_log2,
|
||||
"aten::logaddexp": ops.core.aten_logaddexp,
|
||||
"aten::logaddexp2": ops.core.aten_logaddexp2,
|
||||
"aten::logcumsumexp": ops.core.aten_logcumsumexp,
|
||||
"aten::logdet": ops.core.aten_logdet,
|
||||
"aten::logsigmoid": ops.nn.aten_log_sigmoid,
|
||||
"aten::logsumexp": ops.core.aten_logsumexp,
|
||||
"aten::lt": ops.core.aten_lt,
|
||||
"aten::masked_fill": ops.core.aten_masked_fill,
|
||||
"aten::matmul": ops.core.aten_matmul,
|
||||
"aten::maximum": ops.core.aten_maximum,
|
||||
"aten::minimum": ops.core.aten_minimum,
|
||||
"aten::mm": ops.core.aten_mm,
|
||||
"aten::mul": ops.core.aten_mul,
|
||||
"aten::native_layer_norm": ops.core.aten_native_layer_norm,
|
||||
"aten::ne": ops.core.aten_ne,
|
||||
"aten::neg": ops.core.aten_neg,
|
||||
"aten::new_full": ops.core.aten_new_full,
|
||||
"aten::adaptive_avg_pool1d": ops.nn.aten_adaptive_avg_pool1d,
|
||||
"aten::adaptive_avg_pool2d": ops.nn.aten_adaptive_avg_pool2d,
|
||||
"aten::adaptive_avg_pool3d": ops.nn.aten_adaptive_avg_pool3d,
|
||||
"aten::celu": ops.nn.aten_celu,
|
||||
"aten::elu": ops.nn.aten_elu,
|
||||
"aten::embedding": ops.core.aten_embedding,
|
||||
"aten::gelu": ops.nn.aten_gelu,
|
||||
"aten::leaky_relu": ops.nn.aten_leaky_relu,
|
||||
"aten::linear": ops.nn.aten_linear,
|
||||
"aten::logsigmoid": ops.nn.aten_log_sigmoid,
|
||||
"aten::relu": ops.nn.aten_relu,
|
||||
"aten::relu6": ops.nn.aten_relu6,
|
||||
"aten::selu": ops.core.aten_selu,
|
||||
"aten::upsample_nearest2d": ops.nn.aten_upsample_nearest2d,
|
||||
"aten::nonzero": ops.core.aten_nonzero,
|
||||
"aten::ones_like": ops.core.aten_ones_like,
|
||||
"aten::ones": ops.core.aten_ones,
|
||||
"aten::permute": ops.core.aten_permute,
|
||||
"aten::pow": ops.core.aten_pow,
|
||||
"aten::reciprocal": ops.core.aten_reciprocal,
|
||||
"aten::relu": ops.nn.aten_relu,
|
||||
"aten::relu6": ops.nn.aten_relu6,
|
||||
"aten::remainder": ops.core.aten_remainder,
|
||||
"aten::repeat": ops.core.aten_repeat,
|
||||
"aten::reshape": ops.core.aten_reshape,
|
||||
"aten::round": ops.core.aten_round,
|
||||
"aten::rsqrt": ops.core.aten_rsqrt,
|
||||
"aten::rsub": ops.core.aten_rsub,
|
||||
"aten::select": ops.core.aten_select,
|
||||
"aten::selu": ops.core.aten_selu,
|
||||
"aten::sigmoid": ops.core.aten_sigmoid,
|
||||
"aten::sign": ops.core.aten_sign,
|
||||
"aten::sin": ops.core.aten_sin,
|
||||
@ -153,21 +159,21 @@ _ATENLIB_FUNCTIONS = {
|
||||
"aten::split": ops.core.aten_split,
|
||||
"aten::sqrt": ops.core.aten_sqrt,
|
||||
"aten::sub": ops.core.aten_sub,
|
||||
"aten::sum": ops.core.aten_sum_dim_IntList,
|
||||
"aten::t": ops.core.aten_t,
|
||||
"aten::tan": ops.core.aten_tan,
|
||||
"aten::tanh": ops.core.aten_tanh,
|
||||
"aten::topk": ops.core.aten_topk,
|
||||
"aten::transpose": ops.core.aten_transpose,
|
||||
"aten::unsqueeze": ops.core.aten_unsqueeze,
|
||||
"aten::upsample_nearest2d": ops.nn.aten_upsample_nearest2d,
|
||||
"aten::view": ops.core.aten_view,
|
||||
"aten::where": ops.core.aten_where,
|
||||
"aten::xlogy": ops.special.aten_special_xlogy,
|
||||
"aten::zeros": ops.core.aten_zeros,
|
||||
"aten::zeros_like": ops.core.aten_zeros_like,
|
||||
"aten::native_layer_norm": ops.core.aten_native_layer_norm,
|
||||
"aten::transpose": ops.core.aten_transpose,
|
||||
"aten::sum": ops.core.aten_sum_dim_IntList,
|
||||
"aten::argmin": ops.core.aten_argmin,
|
||||
"aten::argmax": ops.core.aten_argmax,
|
||||
"aten::zeros": ops.core.aten_zeros,
|
||||
"getitem": aten_getitem,
|
||||
"prims::convert_element_type": prims_convert_element_type,
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user