Skip to content

Commit 7705d73

Browse files
authored
DLFW changes 25.01 onwards (#3356)
1 parent 6a9f699 commit 7705d73

File tree

5 files changed

+9
-7
lines changed

5 files changed

+9
-7
lines changed

core/util/Exception.h

+4
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
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>

examples/int8/training/vgg16/requirements.txt

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ tensorboard>=1.14.0
22
protobuf==3.20.*
33
nvidia-pyindex
44
--extra-index-url https://pypi.nvidia.com
5-
pytorch-quantization
65
tqdm
76
nvidia-modelopt
87
--extra-index-url https://pypi.nvidia.com

noxfile.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Use system installed Python packages
88
PYT_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
)

tests/py/ts/ptq/test_ptq_dataloader_calibrator.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
import torch
55
import torch.nn as nn
66
import torch_tensorrt as torchtrt
7+
import torch_tensorrt.ts.ptq as PTQ
78
import torchvision
89
import torchvision.transforms as transforms
9-
import torch_tensorrt.ts.ptq as PTQ
1010
from torch.nn import functional as F
1111
from 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)

tests/py/ts/ptq/test_ptq_trt_calibrator.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os
22
import unittest
33

4+
import tensorrt as trt
45
import torch
56
import torch.nn as nn
67
import torch_tensorrt as torchtrt
@@ -9,14 +10,12 @@
910
from torch.nn import functional as F
1011
from torch_tensorrt.ts.logging import *
1112

12-
import tensorrt as trt
13-
1413

1514
def 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)

0 commit comments

Comments
 (0)