DLFW changes 25.01 onwards

Author: apbose

examples/int8/training/vgg16/requirements.txt

@@ -2,7 +2,6 @@ tensorboard>=1.14.0
 protobuf==3.20.*
 nvidia-pyindex
 --extra-index-url https://pypi.nvidia.com
-pytorch-quantization
 tqdm
 nvidia-modelopt
 --extra-index-url https://pypi.nvidia.com

noxfile.py

@@ -6,7 +6,7 @@
 
 # Use system installed Python packages
 PYT_PATH = (
-    "/usr/local/lib/python3.10/dist-packages"
+    "/usr/local/lib/python3.12/dist-packages"
     if not "PYT_PATH" in os.environ
     else os.environ["PYT_PATH"]
 )
@@ -34,7 +34,7 @@
 # Set epochs to train VGG model for accuracy tests
 EPOCHS = 25
 
-SUPPORTED_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
+SUPPORTED_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10", "3.12"]
 
 nox.options.sessions = [
     "l0_api_tests-" + "{}.{}".format(sys.version_info.major, sys.version_info.minor)
@@ -233,12 +233,15 @@ def run_dynamo_converter_tests(session):
 def run_dynamo_lower_tests(session):
     print("Running Dynamo lowering passes")
     session.chdir(os.path.join(TOP_DIR, "tests/py/dynamo/"))
+    num_workers = "auto"
     tests = ["lowering"]
     for test in tests:
         if USE_HOST_DEPS:
-            session.run_always("pytest", test, env={"PYTHONPATH": PYT_PATH})
+            session.run_always(
+                "pytest", test, "-n", num_workers, env={"PYTHONPATH": PYT_PATH}
+            )
         else:
-            session.run_always("pytest", test)
+            session.run_always("pytest", test, "-n", num_workers)
 
 
 def run_dynamo_partitioning_tests(session):

tests/py/ts/ptq/test_ptq_dataloader_calibrator.py

@@ -4,9 +4,9 @@
 import torch
 import torch.nn as nn
 import torch_tensorrt as torchtrt
+import torch_tensorrt.ts.ptq as PTQ
 import torchvision
 import torchvision.transforms as transforms
-import torch_tensorrt.ts.ptq as PTQ
 from torch.nn import functional as F
 from torch_tensorrt.ts.logging import *
 
@@ -15,7 +15,7 @@ def find_repo_root(max_depth=10):
     dir_path = os.path.dirname(os.path.realpath(__file__))
     for i in range(max_depth):
         files = os.listdir(dir_path)
-        if "WORKSPACE" in files:
+        if "MODULE.bazel" in files:
             return dir_path
         else:
             dir_path = os.path.dirname(dir_path)

tests/py/ts/ptq/test_ptq_trt_calibrator.py

@@ -1,6 +1,7 @@
 import os
 import unittest
 
+import tensorrt as trt
 import torch
 import torch.nn as nn
 import torch_tensorrt as torchtrt
@@ -9,14 +10,12 @@
 from torch.nn import functional as F
 from torch_tensorrt.ts.logging import *
 
-import tensorrt as trt
-
 
 def find_repo_root(max_depth=10):
     dir_path = os.path.dirname(os.path.realpath(__file__))
     for i in range(max_depth):
         files = os.listdir(dir_path)
-        if "WORKSPACE" in files:
+        if "MODULE.bazel" in files:
             return dir_path
         else:
             dir_path = os.path.dirname(dir_path)