#KiSSKr\rSr\"\R S5(a\R R rSr"SS\5rg)NF_CUDAGreenContextTcd^\rSrSrSr\S S\S\S\4Sjj5rS U4Sjjr S U4Sjjr S r U=r $) GreenContextzmWrapper around a CUDA green context. .. warning:: This API is in beta and may change in future releases. num_sms device_idreturncX[(d [S5e[RX5$)zCreate a CUDA green context. Arguments: num_sms (int): The number of SMs to use in the green context. device_id (int, optional): The device index of green context. z1PyTorch was not built with Green Context support!) SUPPORTED RuntimeError _GreenContextcreate)rrs [/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/torch/cuda/green_contexts.pyrGreenContext.creates&yRS S##G77c >[TU]5$)z+Make the green context the current context.)super set_contextself __class__s rrGreenContext.set_context#sw"$$rc >[TU]5$)zsAssuming the green context is the current context, pop it from the context stack and restore the previous context. )r pop_contextrs rrGreenContext.pop_context'sw"$$r)r)r N) __name__ __module__ __qualname____firstlineno____doc__ staticmethodintr rrr__static_attributes__ __classcell__)rs@rrrs@  8 8 8M 8 8%%%rr)torchobjectr r hasattr_Crrrrrr*sF    588())HH..MI %=%r