U <c @sddlZddlZddlmZddlmZmZmZdgZdZ dZ ddZ gZ e e d D]NZ e e e d D]:Ze eZe e ed D]ZeeZee kre eqqfqTe Gd ddejZdS) N) benchmark)FuzzedParameter FuzzedTensorParameterAliasSpectralOpFuzzeri@cs&fddttt|dDS)Nc3s|]}|VqdS)N.0ibaserM/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/benchmark/op_fuzzers/spectral.py szpower_range..)rangeintmathlog)Z upper_boundr rr r power_range srcs2eZdZejdddeeedfddZZ S)rF?)dtypecudaprobability_regular)seedrrcstjtdddddddddtd Dd dtd Dfd dtd Dd dtd Dgtd ddddddd||d g|ddS)Nndimg333333?g?)rrrT) distributionstrictcSs"g|]}td|ttddqS)k_any_Z loguniform)nameminvalmaxvalr)r MIN_DIM_SIZE MAX_DIM_SIZEr rrr -sz-SpectralOpFuzzer.__init__..rcSs(g|] }td|ddtDdqS) k_regular_cSsi|]}|dttqS)r)len REGULAR_SIZES)r sizerrr 8sz8SpectralOpFuzzer.__init__...r"r)rr*r rrrr'5s  c s@g|]8}td|td|td|diddqS)kr(r!rT)r"rr )rrr rrrr';s  c Ss*g|]"}td|dddddddqS)Zstep_g?gQ?g{Gz?)rrrr-)rr rrrr'Gs x)Zk0Zk1Zk2)Zstep_0Zstep_1Zstep_2g?iil) r"r+ZstepsZprobability_contiguousZ min_elementsZ max_elementsZmax_allocation_bytesZ dim_parameterrr) parametersZtensorsr)super__init__rrr)selfrrrr __class__r/rr5s<   .zSpectralOpFuzzer.__init__) __name__ __module__ __qualname__torchZfloat64rboolfloatr5 __classcell__rrr7rrs)rr<Z torch.utilsrZtorch.utils.benchmarkrrr__all__r%r&rr*r jZijr.ZijkappendsortZFuzzerrrrrrs"