rh{ tUddlmZddlmZddlZddlmZddlmZdee fdZ dee fdZ eeeeZ ee d <eeZee d <ej gZe dd e d edeeej(ej*fee efffd Zdej(defdZdeedej(defdZy))Optional) TypeAliasN)Tensor)no_gradreturncDddtjjgS)z:Return the device type list that supports foreach kernels.cudaxputorch_C_get_privateuse1_backend_namem/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/torch/utils/_foreach_utils.py&_get_foreach_kernels_supported_devicesr s E588AAC DDrcJdddddtjjgS)zEReturn the device type list that supports fused kernels in optimizer.mpsr r hpucpur rrr$_get_fused_kernels_supported_devicesrs-     ..0  rTensorListListIndicestensorlistlist with_indicescBtjj||SN)r r "_group_tensors_by_device_and_dtype)rrs rrr*s 88 6 6~| TTrdevicecx|jtdgzvxrtjj S)Nr)typerr jit is_scripting)rs r_device_has_foreach_supportr$2s6 >@E7JK ) &&( (rtensorsc@t|xrtd|DS)Nc3JK|]}|duxst|tvywr)r!_foreach_supported_types).0ts r z'_has_foreach_support..:s+7=>T 8T!W 8887s!#)r$all)r%rs r_has_foreach_supportr-9s( &v . 37BI74r)F)typingrtyping_extensionsrr rtorch.autograd.grad_moderliststrrrr__annotations__intrr(booldicttuplerdtyperr$r-rrrr9s' ,ES E d3i !hv&6!78 8#Y!LL> U"UU % ekk) *E.'2I,J JKU U $v, r