File tree Expand file tree Collapse file tree 1 file changed +0
-3
lines changed
intel_extension_for_pytorch/cpu Expand file tree Collapse file tree 1 file changed +0
-3
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments