f^(h:ddlZddlmZddlmZGddeZy)N)ProxyTorchDispatchMode)TorchFunctionModeceZdZdZddZy)AutogradStateOpsFailSafeguarda Detect grad state ops during exporting the graph and fail the process by raising an error, to avoid unexpected behavior. Those grad mode ops could be: `torch.no_grad` `torch.enable_grad` `torch.set_grad_enabled` Export with predispatch mode is exempted. Nc|xsi}tjjg}tjj}||vrt |dk(sJ|d}tjj tjj j}|r0t|tr |js||k7rtd|d||i|S)Nrz&Encountered autograd state manager op z trying to change global autograd state while exporting. This is unsafe because we don't capture this op in torch.export today, hence we can't reflect the user intention soundly. You can fix this by adding a torch.no_grad() context around the export call.) torch_C_set_grad_enabledis_grad_enabledlen_get_dispatch_mode_TorchDispatchModeKeyPROXY isinstancer pre_dispatch RuntimeError) selffunctypesargskwargsunsupported_grad_mode_ops current_state changed_statemodes U/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/export/_safeguard.py__torch_function__z0AutogradStateOpsFailSafeguard.__torch_function__s2 HH & &% ! 002 , ,t9> !> GM88..uxx/M/M/S/STDt%;<))!]2"r's E-%%$5%%r