File tree Expand file tree Collapse file tree 1 file changed +38
-1
lines changed Expand file tree Collapse file tree 1 file changed +38
-1
lines changed Original file line number Diff line number Diff line change @@ -250,6 +250,43 @@ docker run -it -v$(pwd)/..:/workspace/Torch-TensorRT build_torch_tensorrt_wheel
250
250
251
251
Python compilation expects using the tarball based compilation strategy from above.
252
252
253
+
254
+ ## Testing using Python backend
255
+
256
+ Torch-TensorRT supports testing in Python using [ nox] ( https://nox.thea.codes/en/stable )
257
+
258
+ To install the nox using python-pip
259
+
260
+ ```
261
+ python3 -m pip install --upgrade nox
262
+ ```
263
+
264
+ To list supported nox sessions:
265
+
266
+ ```
267
+ nox --session -l
268
+ ```
269
+
270
+ Environment variables supported by nox
271
+
272
+ ```
273
+ PYT_PATH - To use different PYTHONPATH than system installed Python packages
274
+ TOP_DIR - To set the root directory of the noxfile
275
+ USE_CXX11 - To use cxx11_abi (Defaults to 0)
276
+ USE_HOST_DEPS - To use host dependencies for tests (Defaults to 0)
277
+ ```
278
+
279
+ Usage example
280
+
281
+ ```
282
+ nox --session l0_api_tests
283
+ ```
284
+
285
+ Supported Python versions:
286
+ ```
287
+ ["3.7", "3.8", "3.9", "3.10"]
288
+ ```
289
+
253
290
## How do I add support for a new op...
254
291
255
292
### In Torch-TensorRT?
@@ -279,4 +316,4 @@ Take a look at the [CONTRIBUTING.md](CONTRIBUTING.md)
279
316
280
317
## License
281
318
282
- The Torch-TensorRT license can be found in the LICENSE file. It is licensed with a BSD Style licence
319
+ The Torch-TensorRT license can be found in the LICENSE file. It is licensed with a BSD Style licence
You can’t perform that action at this time.
0 commit comments