File tree 5 files changed +9
-7
lines changed
examples/int8/training/vgg16
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
1
5
#pragma once
2
6
3
7
#include < exception>
Original file line number Diff line number Diff line change @@ -2,7 +2,6 @@ tensorboard>=1.14.0
2
2
protobuf == 3.20.*
3
3
nvidia-pyindex
4
4
--extra-index-url https://pypi.nvidia.com
5
- pytorch-quantization
6
5
tqdm
7
6
nvidia-modelopt
8
7
--extra-index-url https://pypi.nvidia.com
Original file line number Diff line number Diff line change 6
6
7
7
# Use system installed Python packages
8
8
PYT_PATH = (
9
- "/usr/local/lib/python3.10 /dist-packages"
9
+ "/usr/local/lib/python3.12 /dist-packages"
10
10
if not "PYT_PATH" in os .environ
11
11
else os .environ ["PYT_PATH" ]
12
12
)
Original file line number Diff line number Diff line change 4
4
import torch
5
5
import torch .nn as nn
6
6
import torch_tensorrt as torchtrt
7
+ import torch_tensorrt .ts .ptq as PTQ
7
8
import torchvision
8
9
import torchvision .transforms as transforms
9
- import torch_tensorrt .ts .ptq as PTQ
10
10
from torch .nn import functional as F
11
11
from torch_tensorrt .ts .logging import *
12
12
@@ -15,7 +15,7 @@ def find_repo_root(max_depth=10):
15
15
dir_path = os .path .dirname (os .path .realpath (__file__ ))
16
16
for i in range (max_depth ):
17
17
files = os .listdir (dir_path )
18
- if "WORKSPACE " in files :
18
+ if "MODULE.bazel " in files :
19
19
return dir_path
20
20
else :
21
21
dir_path = os .path .dirname (dir_path )
Original file line number Diff line number Diff line change 1
1
import os
2
2
import unittest
3
3
4
+ import tensorrt as trt
4
5
import torch
5
6
import torch .nn as nn
6
7
import torch_tensorrt as torchtrt
9
10
from torch .nn import functional as F
10
11
from torch_tensorrt .ts .logging import *
11
12
12
- import tensorrt as trt
13
-
14
13
15
14
def find_repo_root (max_depth = 10 ):
16
15
dir_path = os .path .dirname (os .path .realpath (__file__ ))
17
16
for i in range (max_depth ):
18
17
files = os .listdir (dir_path )
19
- if "WORKSPACE " in files :
18
+ if "MODULE.bazel " in files :
20
19
return dir_path
21
20
else :
22
21
dir_path = os .path .dirname (dir_path )
You can’t perform that action at this time.
0 commit comments