Skip to content

Commit 9dcdb56

Browse files
authored
fix dlrm ddp regression issue with the launch script (#884)
1 parent f40dbad commit 9dcdb56

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

intel_extension_for_pytorch/cpu/launch.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -621,14 +621,11 @@ def launch(self, args):
621621
self.set_env("MASTER_PORT", str(args.master_port))
622622
mpi_pin_domain = self.get_mpi_pin_domain(args.nproc_per_node, args.ccl_worker_count, total_cores_per_node)
623623
self.set_env("I_MPI_PIN_DOMAIN", mpi_pin_domain)
624-
mpi_pin_domain = os.environ["I_MPI_PIN_DOMAIN"]
625624

626625
ppn = args.nproc_per_node
627626
cores_per_rank = total_cores_per_node // ppn
628627

629628
omp_num_threads = cores_per_rank - args.ccl_worker_count
630-
self.set_env("OMP_NUM_THREADS", str(omp_num_threads))
631-
omp_num_threads = os.environ["OMP_NUM_THREADS"]
632629
self.set_multi_thread_and_allocator(omp_num_threads,
633630
args.disable_iomp,
634631
True,

0 commit comments

Comments
 (0)