o lÛ‚iã@s<ddlZeZdZeejdƒrejjZdZGdd„deƒZdS)éNFÚ_CUDAGreenContextTcsLeZdZdZed dededefdd„ƒZd‡fd d „ Zd‡fd d „ Z ‡Z S)Ú GreenContextzyWrapper around a CUDA green context. .. warning:: This API is in beta and may change in future releases. rÚnum_smsÚ device_idÚreturncCststdƒ‚t ||¡S)zÍCreate 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Ú _GreenContextÚcreate)rr©r ú\/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/torch/cuda/green_contexts.pyr s zGreenContext.createNcó tƒ ¡S)z+Make the green context the current context.)ÚsuperÚ set_context©Úself©Ú __class__r r r#s zGreenContext.set_contextcr )zƒAssuming the green context is the current context, pop it from the context stack and restore the previous context. )rÚ pop_contextrrr r r's zGreenContext.pop_context)r)rN) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú staticmethodÚintr r rrÚ __classcell__r r rr rs  r)ÚtorchÚobjectr rÚhasattrÚ_Crrr r r r Ús