fixed an affinity bug (#363)

* fixed an affinity bug

* bug fix in test_launcher.py

Author: jingxu10

intel_extension_for_pytorch/cpu/launch.py

@@ -457,6 +457,7 @@ def get_ccl_worker_affinity(self, nproc_per_node, ccl_worker_count, total_cores)
         for proc in range(ppn):
             for ccl_worker in range(ccl_worker_count):
                 affinity += str(proc * cores_per_rank + ccl_worker) + ","
+        affinity = affinity[:-1]
         return affinity
 
     def launch(self, args):

tests/cpu/test_launcher.py

@@ -16,6 +16,8 @@ def find_lib(self, lib_type):
         library_paths = []
         if "CONDA_PREFIX" in os.environ:
             library_paths.append(os.environ["CONDA_PREFIX"] + "/lib/")
+        elif "VIRTUAL_ENV" in os.environ:
+            library_paths.append(os.environ["VIRTUAL_ENV"] + "/lib/")
 
         library_paths += ["{}/.local/lib/".format(expanduser("~")), "/usr/local/lib/",
                          "/usr/local/lib64/", "/usr/lib/", "/usr/lib64/"]
@@ -63,7 +65,7 @@ def test_mpi_pin_domain_and_ccl_worker_affinity(self):
        expect_pin_domain = "[0xffffff0,0xffffff00000000,]"
        self.assertEqual(pin_doamin, expect_pin_domain)
        ccl_worker_affinity = launcher.get_ccl_worker_affinity(proc_per_node, ccl_worker_count, total_cores)
-       expect_ccl_worker_affinity = "0,1,2,3,28,29,30,31,"
+       expect_ccl_worker_affinity = "0,1,2,3,28,29,30,31"
        self.assertEqual(ccl_worker_affinity, expect_ccl_worker_affinity)