thunder.core.prims.PrimIDs¶

class thunder.core.prims.PrimIDs(value)[source]¶

Bases: Enum

An enumeration.

__init__()[source]¶

Attributes

`ASSERT_TENSOR_METADATA`
`CHECK_TENSOR_SHAPE_AND_METADATA`
`CHECK_NONE`
`CHECK_EMPTY`
`CHECK_LITERAL_LIKE`
`CHECK_TYPE`
`CHECK_INSTANCE`
`CHECK_NUMBER_TYPE_AND_VALUE`
`CHECK_BOOL_CONVERSION`
`CHECK_STRING_VALUE`
`CHECK_SLICE_VALUE`
`CHECK_LEN`
`ASSERT_COMPARE`
`PYTHON_VARS`
`UNPACK_FUNCTION_OBJ`
`UNPACK_CACHE_INFO`
`UNPACK_ATTR`
`UNPACK_GETITEM`
`UNPACK_EMPTY_DICT`
`UNPACK_ITER`
`UNPACK_NEXT`
`UNPACK_KEY`
`UNPACK_SEQUENCE`
`UNPACK_TRIVIAL`
`UNPACK_TUPLE`
`UNPACK_LIST`
`UNPACK_DICT_KEY`
`UNPACK_PARAMETER`
`UNPACK_BUFFER`
`UNPACK_SUBMODULE`
`UNPACK_THUNDER_MODULE`
`CONSTRUCT_TUPLE`
`PACK_LIST`
`PACK_BUFFER`
`PACK_ATTR`
`PACK_SETITEM`
`DATACLASS_NEW`
`SHAPE`
`COMMENT`
`DEL`
`PRINT`
`RETURN`
`GET_GRAD`
`PUT_GRAD`
`CONVERT_ELEMENT_TYPE`
`DEVICE_PUT`
`NUMPY_ARRAY_TO_TORCH_TENSOR`
`EXOGENOUS_LIKE`
`FULL`
`IOTA`
`UNIFORM`
`UNIFORM_PHILOX`
`RANDINT`
`RANDN`
`EMPTY`
`TENSOR_FROM_SEQUENCE`
`CLONE`
`UPDATE_ALIASES`
`MULTINOMIAL`
`GET_AND_UPDATE_RNG_STATE`
`BROADCAST_IN_DIM`
`CAT`
`FLIP`
`RESHAPE`
`SLICE`
`SQUEEZE`
`TRANSPOSE`
`UNFOLD`
`VIEW`
`SHALLOW_COPY`
`STRIDE_ORDER`
`PY_ABS`
`ABS`
`ACOS`
`ACOSH`
`ASIN`
`ASINH`
`ATAN`
`ATANH`
`BITWISE_NOT`
`CEIL`
`COS`
`COSH`
`DIGAMMA`
`ERF`
`ERFC`
`ERFCINV`
`ERFINV`
`EXP`
`EXP2`
`EXPM1`
`FLOOR`
`FREXP`
`ISFINITE`
`LGAMMA`
`LOG`
`LOG10`
`LOG1P`
`LOG2`
`NDTRI`
`NEG`
`RECIPROCAL`
`ROUND`
`RSQRT`
`SIGN`
`SIGNBIT`
`SIN`
`SINH`
`SQRT`
`TAN`
`TANH`
`TRUNC`
`REAL`
`IMAG`
`ADD`
`ATAN2`
`BITWISE_AND`
`BITWISE_OR`
`BITWISE_XOR`
`DIV`
`EQ`
`PY_FLOORDIV`
`FMOD`
`GE`
`GT`
`LE`
`LT`
`MAXIMUM`
`MINIMUM`
`MUL`
`NE`
`NEXTAFTER`
`POW`
`REMAINDER`
`SUB`
`ZETA`
`BITWISE_LEFT_SHIFT`
`BITWISE_RIGHT_SHIFT`
`LERP`
`WHERE`
`AMAX`
`AMIN`
`PROD`
`SUM`
`VAR`
`VAR_MEAN`
`STD`
`ARGMAX`
`ARGMIN`
`TOPK`
`SORT`
`GATHER`
`SCATTER`
`INDEX_ADD`
`INDEX_COPY`
`INDEX_PUT`
`SCATTER_ADD`
`TAKE`
`TAKE_ALONG_AXIS`
`COPY_WITH_SETITEM`
`MATMUL`
`CONVOLUTION`
`EMBEDDING`
`EMBEDDING_BACKWARD`
`LINEAR`
`PAD`
`ITEM`
`COPY_`
`BITCAST`
`SINK`