up

Author: metascroy

install_requirements.py

@@ -10,6 +10,7 @@
 import re
 import subprocess
 import sys
+import os
 
 
 def python_is_compatible():
@@ -117,6 +118,8 @@ def install_requirements(use_pytorch_nightly):
 
     # Install packages directly from local copy instead of pypi.
     # This is usually not recommended.
+    new_env = os.environ.copy()
+    new_env["USE_CPP"] = "1" # build torchao kernels
     subprocess.run(
         [
             sys.executable,
@@ -127,6 +130,7 @@ def install_requirements(use_pytorch_nightly):
             "--no-build-isolation",
             *LOCAL_REQUIREMENTS,
         ],
+        env=new_env,
         check=True,
     )