Compiler options for windows env (apache#49)

Co-authored-by: Siva <quic_sivb@quicinc.com>
krishnaraj36 · Aug 8, 2024 · 7dd77e6 · 7dd77e6
1 parent f09a0aa
commit 7dd77e6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/tvm/contrib/cc.py b/python/tvm/contrib/cc.py
@@ -372,8 +372,11 @@ def _linux_compile(
 
 
 def _windows_compile(output, objects, options, cwd=None, ccache_env=None):
-    cmd = ["clang"]
-    cmd += ["-O2 --target=x86_64"]
+    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"]