iddlZeZdZeej drej j ZdZGddeZy)NF_CUDAGreenContextTcPeZdZdZeddededefdZd fd Zd fd Z xZ S) GreenContextzyWrapper around a CUDA green context. .. warning:: This API is in beta and may change in future releases. num_sms device_idreturncPts tdtj||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 S/var/www/html/engine/venv/lib/python3.12/site-packages/torch/cuda/green_contexts.pyr zGreenContext.creates&RS S##GY77c t|S)z+Make the green context the current context.)super set_contextself __class__s rrzGreenContext.set_context#sw"$$rc t|S)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'sw"$$r)r)rN) __name__ __module__ __qualname____doc__ staticmethodintr r rr __classcell__)rs@rrrs@  8 8 8M 8 8%%%rr)torchobjectr r hasattr_Crrrrr$sC    588()HH..MI %=%r