P/jRxUddlmZddlmZddlmZdaeeje d<deejfdZ y))OptionalN)config _COMPILE_PGreturnctjrTtjr@tjr,t tj jdat Sy)Npt2_compile_pg)pg_tag)renable_compiler_collectivesdist is_availableis_initializedrdistributed_c10d_new_group_with_tag\/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/_dynamo/distributed.pyget_compile_pgr sU**        !  //CC'DK r) typingrtorch.distributed distributedr rr ProcessGroup__annotations__rrrrrs< ,0 Xd'' (/!2!23r