NjSSKr\rSr\"\R S5(a\R R rSr"SS\5rg)NF_CUDAGreenContextTc^\rSrSrSr\S S\S\S\4Sjj5rS U4Sjjr S U4Sjjr S\ R4U4S jjr 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 j/root/GenerationalWealth/GenerationalWealth/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"$$rc >[TU]5$)z1Return the CUDA Stream used by the green context.)rStreamrs rrGreenContext.Stream-sw~r)r)r N)__name__ __module__ __qualname____firstlineno____doc__ staticmethodintr rrrtorchr__static_attributes__ __classcell__)rs@rrrsP  8 8 8M 8 8%%     rr)r'objectr r hasattr_Crrrrrr-sF    588())HH..MI ! =! r