File tree
5 files changed
+32
-7
lines changed- cpp/include/torch_tensorrt
- py/torch_tensorrt/ts
5 files changed
+32
-7
lines changedDiff for: .pre-commit-config.yaml
+3-3
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
1 | 1 |
| |
2 | 2 |
| |
3 | 3 |
| |
4 |
| - | |
| 4 | + | |
5 | 5 |
| |
6 | 6 |
| |
7 | 7 |
| |
| |||
32 | 32 |
| |
33 | 33 |
| |
34 | 34 |
| |
35 |
| - | |
| 35 | + | |
36 | 36 |
| |
37 | 37 |
| |
38 | 38 |
| |
| |||
52 | 52 |
| |
53 | 53 |
| |
54 | 54 |
| |
55 |
| - | |
| 55 | + | |
56 | 56 |
| |
57 | 57 |
| |
58 | 58 |
| |
|
Diff for: cpp/include/torch_tensorrt/macros.h
+1
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
30 | 30 |
| |
31 | 31 |
| |
32 | 32 |
| |
| 33 | + | |
33 | 34 |
| |
34 | 35 |
| |
35 | 36 |
| |
|
Diff for: cpp/include/torch_tensorrt/ptq.h
+3-4
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
59 | 59 |
| |
60 | 60 |
| |
61 | 61 |
| |
62 |
| - | |
| 62 | + | |
63 | 63 |
| |
64 | 64 |
| |
65 | 65 |
| |
| |||
308 | 308 |
| |
309 | 309 |
| |
310 | 310 |
| |
311 |
| - | |
312 | 311 |
| |
313 |
| - | |
| 312 | + | |
314 | 313 |
| |
315 | 314 |
| |
316 | 315 |
| |
| |||
344 | 343 |
| |
345 | 344 |
| |
346 | 345 |
| |
347 |
| - | |
| 346 | + | |
348 | 347 |
| |
349 | 348 |
| |
350 | 349 |
| |
|
Diff for: py/torch_tensorrt/ts/_compiler.py
+13
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
1 | 1 |
| |
2 | 2 |
| |
| 3 | + | |
3 | 4 |
| |
4 | 5 |
| |
5 | 6 |
| |
| |||
102 | 103 |
| |
103 | 104 |
| |
104 | 105 |
| |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
105 | 112 |
| |
106 | 113 |
| |
107 | 114 |
| |
| |||
240 | 247 |
| |
241 | 248 |
| |
242 | 249 |
| |
| 250 | + | |
| 251 | + | |
| 252 | + | |
| 253 | + | |
| 254 | + | |
| 255 | + | |
243 | 256 |
| |
244 | 257 |
| |
245 | 258 |
| |
|
Diff for: py/torch_tensorrt/ts/ptq.py
+12
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
13 | 13 |
| |
14 | 14 |
| |
15 | 15 |
| |
| 16 | + | |
| 17 | + | |
16 | 18 |
| |
17 | 19 |
| |
18 | 20 |
| |
| |||
88 | 90 |
| |
89 | 91 |
| |
90 | 92 |
| |
| 93 | + | |
| 94 | + | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
91 | 98 |
| |
92 | 99 |
| |
93 | 100 |
| |
| |||
175 | 182 |
| |
176 | 183 |
| |
177 | 184 |
| |
| 185 | + | |
| 186 | + | |
| 187 | + | |
| 188 | + | |
| 189 | + | |
178 | 190 |
| |
179 | 191 |
| |
180 | 192 |
| |
|
0 commit comments