Searched refs:enable_compiler_collectives (Results 1 – 3 of 3) sorted by relevance
495 @config.patch(optimize_ddp=False, enable_compiler_collectives=True)514 @config.patch(optimize_ddp=True, enable_compiler_collectives=True)522 @config.patch(optimize_ddp=True, enable_compiler_collectives=True)535 @config.patch(optimize_ddp=True, enable_compiler_collectives=True)541 @config.patch(optimize_ddp=True, enable_compiler_collectives=True)547 @config.patch(optimize_ddp=False, enable_compiler_collectives=True)589 @config.patch(enable_compiler_collectives=True)613 @config.patch(enable_compiler_collectives=True)630 @config.patch(enable_compiler_collectives=True)655 @config.patch(enable_compiler_collectives=True)[all …]
13 config.enable_compiler_collectives
478 enable_compiler_collectives = os.environ.get("TORCH_COMPILER_COLLECTIVES", "0") == "1" variable