U <c @stddlZddlZddlmZmZmZmZdZdZ e dde e e ee e e dDZGdd d eZdS) N)FuzzerFuzzedParameterParameterAliasFuzzedSparseTensoriccs|]}d|VqdS)N.0irrQ/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/benchmark/op_fuzzers/sparse_unary.py sr cs$eZdZejdffdd ZZS)UnaryOpSparseFuzzerFcstjtdddddddtddddddd d d td Dd d td Ddd td Dtddddddtdddddtdddddgtdddddddd||d g|ddS) N dim_parameterg333333?g?)rrT) distributionstrict sparse_dim皙?namerrcSs"g|]}td|ttddqS)k_any_Z loguniformrminvalmaxvalr)r _MIN_DIM_SIZE _MAX_DIM_SIZEr rrr !sz0UnaryOpSparseFuzzer.__init__..rcSs(g|] }td|ddtDdqS)k_pow2_cSsi|]}|dttqS)g?)len_POW_TWO_SIZES)r sizerrr ,sz;UnaryOpSparseFuzzer.__init__...rr)rr!r rrr r)s  c Ss<g|]4}td|td|dtd|diddqS)krg?rrTr)rrr rrr r/s  density)g?g?g{Gz?r$ coalescedg?)TFZ random_valuerluniformrx)Zk0Zk1Zk2ii) rr"rrZ min_elementsZ max_elementsr&r'dtypecuda) parametersZtensorsseed)super__init__rranger)selfr-r*r+ __class__rr r/sR    3zUnaryOpSparseFuzzer.__init__)__name__ __module__ __qualname__torchZfloat32r/ __classcell__rrr2r rsr)Znumpynpr7Ztorch.utils.benchmarkrrrrrrtupler0intlog2r!rrrrr s