a d@s\ddlZddlmZmZddlmZmZddlmZm Z m Z m Z m Z ddl Z ddlmZddlmZd!ee jddd Zd"e e ed d d Ze e ee e je jfd ddZe je jdddZe jeeee jdddZe je e je je jfdddZeeedddZe je je e je je jfdddZe je e je jfddd ZdS)#N)partialwraps)isclass isfunction)AnyCallableListOptionalTuple)Tensor)torch_version_geq)indexreturnc Cs^z*tjrtd|}n td}Wn.tyX}ztd}WYd}~n d}~00|S)zTries to get cuda device, if fail, returns cpu. Args: index: cuda device index Returns: torch.device zcuda:cpuN)torchZcudaZ is_availabledevice BaseException)r dever]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/utils/helpers.pyget_cuda_device_if_available s   rfunc replace_withcs.durttdStfdd}|S)N)rcsld}trjj}tr"j}durHtjd|ddtdntjd|dtd|i|S)N`z` is deprecated in favor of `z`.)categoryz;` is deprecated and will be removed in the future versions.)r __class____name__rwarningswarnDeprecationWarning)argskwargsnamerrrwrapper#s z_deprecated..wrapper)r _deprecatedr)rrr&rrrr's  r') tensor_listrc Csd\}}|D]t}|dur t|tjfs(q |j}|j}|durN|durN|}|}q ||ks^||kr td|d|d|d|d q |durtd}|durt}||fS)zCheck if all the input are in the same device (only if when they are torch.Tensor). If so, it would return a tuple of (device, dtype). Default: (cpu, ``get_default_dtype()``). Returns: [torch.device, torch.dtype] )NNNz8Passed values are not in the same device and dtype.Got (z, z) and (z).r) isinstancerr rdtype ValueErrorZget_default_dtype)r(rr*ZtensorZ_deviceZ_dtyperrr_extract_device_dtype5s6 r,)inputrcCsTt|tjs tdt|d|j}|tjtjfvrt||}tdt||tjt|tj|j dfSt |tj s^t dt |d|j}|tjtjfvrztj}tj||dd\}}}|d k}t||||}||j||j|fS) zsHelper function, which avoids crashing because of singular matrix input and outputs the mask of valid solution. z>PyTorch version < 1.10, solve validness mask maybe not correctr*rzB must be torch.Tensor. Got: r/T)Z get_infosr)r r?r r!RuntimeWarningroneslenboolrr)r r0r1r*r2r3Zlur5Zlu_solve) r=r<Zsolr*ZA_LUZpivotsinfoZ valid_maskXrrrsafe_solve_with_masks    rI)r<rcCstdds:t|}tdt|tjt|tj|j dfSt |tj sZt dt |d|j}|tjtjfvrxtj}n|}ddlm}|||\}}|dk}|||fS) zyHelper function, which avoids crashing because of non-invertable matrix input and outputs the mask of valid solution.r@ z?PyTorch version < 1.9, inverse validness mask maybe not correctrBzA must be torch.Tensor. Got: r/r)inv_ex)r r6r r!rCrrDrErFrr)r r0r1r*r2r3Z torch.linalgrKr5)r<invZdtype_originalr*rKr4rGmaskrrrsafe_inverse_with_masks    rN)r)NN)r functoolsrrinspectrrtypingrrrr r rZ kornia.corer Zkornia.utils._compatr intrrstrr'r*r,r6r:r;r?rIrNrrrrs  $"&