9ipddlZeZdZeejdrejjZdZGddeZdS)NF_CUDAGreenContextTcVeZdZdZed dededefdZd fd Zd fd Z xZ S) GreenContextzyWrapper around a CUDA green context. .. warning:: This API is in beta and may change in future releases. rnum_sms device_idreturncdtstdt||S)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 L/var/www/icac/venv/lib/python3.11/site-packages/torch/cuda/green_contexts.pyr zGreenContext.creates2 TRSS S##GY777NcDtS)z+Make the green context the current context.)super set_contextself __class__s rrzGreenContext.set_context#sww""$$$rcDtS)zAssuming the green context is the current context, pop it from the context stack and restore the previous context. )r pop_contextrs rrzGreenContext.pop_context'sww""$$$r)r)rN) __name__ __module__ __qualname____doc__ staticmethodintr r rr __classcell__)rs@rrrs  8 8 8 8M 8 8 8\ 8%%%%%%%%%%%%%%%%rr)torchobjectr r hasattr_Crrrrr$st    758())H.MI %%%%%=%%%%%r