File tree Expand file tree Collapse file tree 5 files changed +9
-7
lines changed
examples/int8/training/vgg16 Expand file tree Collapse file tree 5 files changed +9
-7
lines changed Original file line number Diff line number Diff line change 1+ // Include cstdint for GCC 13+ or Clang 13+
2+ #if (defined(__GNUC__) && !defined(__clang__) && (__GNUC__ >= 13)) || (defined(__clang__) && (__clang_major__ >= 13))
3+ #include < cstdint>
4+ #endif
15#pragma once
26
37#include < exception>
Original file line number Diff line number Diff line change @@ -2,7 +2,6 @@ tensorboard>=1.14.0
22protobuf == 3.20.*
33nvidia-pyindex
44--extra-index-url https://pypi.nvidia.com
5- pytorch-quantization
65tqdm
76nvidia-modelopt
87--extra-index-url https://pypi.nvidia.com
Original file line number Diff line number Diff line change 66
77# Use system installed Python packages
88PYT_PATH = (
9- "/usr/local/lib/python3.10 /dist-packages"
9+ "/usr/local/lib/python3.12 /dist-packages"
1010 if not "PYT_PATH" in os .environ
1111 else os .environ ["PYT_PATH" ]
1212)
Original file line number Diff line number Diff line change 44import torch
55import torch .nn as nn
66import torch_tensorrt as torchtrt
7+ import torch_tensorrt .ts .ptq as PTQ
78import torchvision
89import torchvision .transforms as transforms
9- import torch_tensorrt .ts .ptq as PTQ
1010from torch .nn import functional as F
1111from torch_tensorrt .ts .logging import *
1212
@@ -15,7 +15,7 @@ def find_repo_root(max_depth=10):
1515 dir_path = os .path .dirname (os .path .realpath (__file__ ))
1616 for i in range (max_depth ):
1717 files = os .listdir (dir_path )
18- if "WORKSPACE " in files :
18+ if "MODULE.bazel " in files :
1919 return dir_path
2020 else :
2121 dir_path = os .path .dirname (dir_path )
Original file line number Diff line number Diff line change 11import os
22import unittest
33
4+ import tensorrt as trt
45import torch
56import torch .nn as nn
67import torch_tensorrt as torchtrt
910from torch .nn import functional as F
1011from torch_tensorrt .ts .logging import *
1112
12- import tensorrt as trt
13-
1413
1514def find_repo_root (max_depth = 10 ):
1615 dir_path = os .path .dirname (os .path .realpath (__file__ ))
1716 for i in range (max_depth ):
1817 files = os .listdir (dir_path )
19- if "WORKSPACE " in files :
18+ if "MODULE.bazel " in files :
2019 return dir_path
2120 else :
2221 dir_path = os .path .dirname (dir_path )
You can’t perform that action at this time.
0 commit comments