U <c @stddlZddlZddlmZmZmZmZdZdZ e dde e e ee e e dDZGdd d eZdS) N)FuzzerFuzzedParameterParameterAlias FuzzedTensoriccs|]}d|VqdS)N.0irrK/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/benchmark/op_fuzzers/binary.py sr cs$eZdZejdffdd ZZS)BinaryOpFuzzerFcstjtdddddddddtd Dd dtd Dd dtd Dd dtd Dd dtd Dtdddddgtdddddddd||d tdddddd||dg|ddS)NZdimg333333?g?)rrT) distributionstrictcSs"g|]}td|ttddqS)k_any_Z loguniformnameminvalmaxvalr)r _MIN_DIM_SIZE _MAX_DIM_SIZEr rrr sz+BinaryOpFuzzer.__init__..rcSs(g|] }td|ddtDdqS)k_pow2_cSsi|]}|dttqS)g?)len_POW_TWO_SIZES)r sizerrr +sz6BinaryOpFuzzer.__init__...rr)rrr rrr r(s  c Ss<g|]4}td|td|dtd|diddqS)kr皙?r皙?Trrrrrr rrr r.s  cSs2g|]*}td|td|dddiddqS)Zy_kr!r"rr#Tr$r%r rrr r9s c Ss8g|]0}dD]&}t|d|dddddddq qS))xyZ_step_r"gQ?g{Gz?)rrrr )r)r r rrrr rEs Z random_valuerluniformrr&)Zk0Zk1Zk2)Zx_step_0Zx_step_1Zx_step_2g?iil) rrstepsprobability_contiguousZ min_elementsZ max_elementsmax_allocation_bytes dim_parameterdtypecudar')Zy_k0Zy_k1Zy_k2)rrr+r,r-r.r/r0) parametersZtensorsseed)super__init__rranger)selfr2r/r0 __class__rr r4sX    @ zBinaryOpFuzzer.__init__)__name__ __module__ __qualname__torchZfloat32r4 __classcell__rrr7r rsr)Znumpynpr<Ztorch.utils.benchmarkrrrrrrtupler5intlog2rrrrrr s