pi 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_gradreturncFdddtjjgS)z:Return the device type list that supports foreach kernels.cudaxpumtiatorch_C_get_privateuse1_backend_name\/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/utils/_foreach_utils.py&_get_foreach_kernels_supported_devicesr s E6588#I#I#K LLrcLddddddtjjgS)zEReturn the device type list that supports fused kernels in optimizer.mpsr r hpucpur r rrr$_get_fused_kernels_supported_devicesrs0     ..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)r s r_device_has_foreach_supportr%3s6 >@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&r s r_has_foreach_supportr.:s( &v . 37BI74r)F)typingrtyping_extensionsrr rtorch.autograd.grad_moderliststrrrr__annotations__intrr)booldicttupler dtyperr%r.rrrr:s' ,MS M d3i !hv&6!78 8#Y!LL> U"UU % ekk) *E.'2I,J JKU U $v, r