File tree Expand file tree Collapse file tree 2 files changed +24
-2
lines changed Expand file tree Collapse file tree 2 files changed +24
-2
lines changed Original file line number Diff line number Diff line change @@ -1597,3 +1597,22 @@ workflows:
1597
1597
trt-version-short : << pipeline.parameters.trt-version-short >>
1598
1598
cudnn-version : << pipeline.parameters.cudnn-version >>
1599
1599
python-version : << pipeline.parameters.python-version >>
1600
+
1601
+ # Temporary, for CI run
1602
+ - build-x86_64-linux :
1603
+ name : build-x86_64-linux-legacy
1604
+ torch-build : << pipeline.parameters.torch-build-legacy >>
1605
+ torchvision-build : << pipeline.parameters.torchvision-build-legacy >>
1606
+ torch-build-index : << pipeline.parameters.torch-build-index-legacy >>
1607
+ python-version : << pipeline.parameters.python-version >>
1608
+ legacy : << pipeline.parameters.enable-legacy >>
1609
+
1610
+ - test-py-ts-x86_64-linux :
1611
+ name : test-py-ts-x86_64-linux-legacy
1612
+ torch-build : << pipeline.parameters.torch-build-legacy >>
1613
+ torchvision-build : << pipeline.parameters.torchvision-build-legacy >>
1614
+ torch-build-index : << pipeline.parameters.torch-build-index-legacy >>
1615
+ trt-version-long : << pipeline.parameters.trt-version-long >>
1616
+ python-version : << pipeline.parameters.python-version >>
1617
+ requires :
1618
+ - build-x86_64-linux-legacy
Original file line number Diff line number Diff line change 1
- from torch_tensorrt .dynamo import fx_ts_compat
2
- from .backend import compile
1
+ import torch
2
+
3
+ if not torch .__version__ .startswith ("1" ):
4
+ from torch_tensorrt .dynamo import fx_ts_compat
5
+ from .backend import compile
You can’t perform that action at this time.
0 commit comments