1c6f1efa8b297f73bde09f8bcbc876439b8edf4e,tools/nni_gpu_tool/gpu_metrics_collector.py,,check_ready_to_run,#,13

Before Change

        pgrep_output = subprocess.check_output("pgrep -fxu "$(whoami)" \"python3 -m nni_gpu_tool.gpu_metrics_collector\"", shell=True)
        pidList = []
        for pid in pgrep_output.splitlines():
            pidList.append(int(pid))
        pidList.remove(os.getpid())
        return not pidList

After Change

        pidList = []
        for pid in pgrep_output.splitlines():
            pid = pid.decode()
            if "pgrep " in pid or pid.startswith("%s " % os.getpid()) or pid.startswith("%s " % os.getppid()):
                continue
            pidList.append(pid)
        return not pidList

Instances