File tree Expand file tree Collapse file tree 2 files changed +22
-1
lines changed
py/torch_tensorrt/dynamo/backend Expand file tree Collapse file tree 2 files changed +22
-1
lines changed Original file line number Diff line number Diff line change @@ -1597,3 +1597,22 @@ workflows:
15971597 trt-version-short : << pipeline.parameters.trt-version-short >>
15981598 cudnn-version : << pipeline.parameters.cudnn-version >>
15991599 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 22from typing import Sequence
33import torch
44from functools import partial
5- import torch ._dynamo as td
5+
6+ if not torch .__version__ .startswith ("1" ):
7+ import torch ._dynamo as td
68
79from torch_tensorrt .dynamo .backend ._settings import CompilationSettings
810from torch_tensorrt .dynamo .backend .lowering ._decompositions import (
You can’t perform that action at this time.
0 commit comments