U <c/ @stddlZddlZddlmZmZmZmZdZdZ e dde e e ee e e dDZGdd d eZdS) N)FuzzerFuzzedParameterParameterAlias FuzzedTensoriccs|]}d|VqdS)N.0irrJ/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/benchmark/op_fuzzers/unary.py sr cs$eZdZejdffdd ZZS) UnaryOpFuzzerFcstjtdddddddddtd Dd dtd Dd dtd Dd dtd Dtd ddddgtdddddddd||d g|ddS)NZdimg333333?g?)rrT) distributionstrictcSs"g|]}td|ttddqS)k_any_Z loguniformnameminvalmaxvalr)r _MIN_DIM_SIZE _MAX_DIM_SIZEr rrr sz*UnaryOpFuzzer.__init__..rcSs(g|] }td|ddtDdqS)k_pow2_cSsi|]}|dttqS)g?)len_POW_TWO_SIZES)r sizerrr )sz5UnaryOpFuzzer.__init__...rr)rrr rrr r&s  c Ss<g|]4}td|td|dtd|diddqS)kr皙?rg?T)rrr)rrr rrr r,s  c Ss*g|]"}td|dddddddqS)Zx_step_r"gQ?g{Gz?)rrrr )rr rrr r8s Z random_valuerluniformrx)Zk0Zk1Zk2)Zx_step_0Zx_step_1Zx_step_2g?iil) rrZstepsZprobability_contiguousZ min_elementsZ max_elementsZmax_allocation_bytesZ dim_parameterdtypecuda) parametersZtensorsseed)super__init__rranger)selfr*r'r( __class__rr r,s>  1zUnaryOpFuzzer.__init__)__name__ __module__ __qualname__torchZfloat32r, __classcell__rrr/r rsr)Znumpynpr4Ztorch.utils.benchmarkrrrrrrtupler-intlog2rrrrrr s