Function _windows_compile

python/tvm/support/cc.py:374–418

(output, objects, options, cwd=None, ccache_env=None)

def _windows_compile(output, objects, options, cwd=None, ccache_env=None):
    compiler = os.getenv("TVM_WIN_CC", default="clang")
    win_target = os.getenv("TVM_WIN_TARGET", default="x86_64")
    cmd = [compiler]
    cmd += ["-O2"]
    cmd += ["--target=" + win_target]

    if output.endswith(".so") or output.endswith(".dll"):
        cmd += ["-shared"]
    elif output.endswith(".obj"):
        cmd += ["-c"]

    if isinstance(objects, str):
        objects = [objects]
    cmd += ["-o", output]
    cmd += objects
    if options:
        cmd += options
    env = None
    if ccache_env is not None:
        if shutil.which("ccache"):
            cmd.insert(0, "ccache")
            env = os.environ.copy()
            env.update(ccache_env)
        else:
            raise ValueError("ccache not found")

    try:
        proc = subprocess.Popen(
            cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, cwd=cwd, env=env
        )
        (out, _) = proc.communicate()
    except FileNotFoundError:
        raise RuntimeError(
            "Can not find the LLVM clang for Windows clang.exe)."
            "Make sure it's installed"
            " and the installation directory is in the %PATH% environment "
            "variable. Prebuilt binaries can be found at: https://llvm.org/"
        )
    if proc.returncode != 0:
        msg = "Compilation error:\n"
        msg += " ".join(cmd) + "\n"
        msg += out.decode("utf-8", errors="replace")

        raise RuntimeError(msg)