acc311dd3e15135ca35f69b0183e5b0f9f4beadf,tools/nni_gpu_tool/gpu_metrics_collector.py,,check_ready_to_run,#,27

Before Change

def check_ready_to_run():
    //TODO check process in windows
    if sys.platform == "win32":
        return True
    pgrep_output =subprocess.check_output("pgrep -fx \"python3 -m nni_gpu_tool.gpu_metrics_collector\"", shell=True)
    pidList = []
    for pid in pgrep_output.splitlines():
        pidList.append(int(pid))

After Change

from xml.dom import minidom

def check_ready_to_run():
    if sys.platform == "win32":
        pgrep_output = subprocess.check_output("wmic process where "CommandLine like \"%nni_gpu_tool.gpu_metrics_collector%\" and name like \"%python%\"" get processId")
        pidList = pgrep_output.decode("utf-8").strip().split()
        pidList.pop(0) // remove the key word "ProcessId"
        pidList = list(map(int, pidList))
        pidList.remove(os.getpid())
        return len(pidList) == 0
    else:
        pgrep_output =subprocess.check_output("pgrep -fx \"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 len(pidList) == 0

def main(argv):
    metrics_output_dir = os.environ["METRIC_OUTPUT_DIR"]
    if check_ready_to_run() == False:
        // GPU metrics collector is already running. Exit

Instances