Skip to content

Commit

Permalink
tool: Auto refresh the lock file on commit
Browse files Browse the repository at this point in the history
Signed-off-by: Naren Dasan <[email protected]>
  • Loading branch information
narendasan committed Sep 18, 2024
1 parent ab14920 commit c9a1690
Show file tree
Hide file tree
Showing 5 changed files with 419 additions and 361 deletions.
6 changes: 6 additions & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,12 @@ repos:
rev: v1.22.9
hooks:
- id: typos
- repo: https://github.com/astral-sh/uv-pre-commit
# uv version.
rev: 0.4.10
hooks:
# Update the uv lockfile
- id: uv-lock
- repo: local
hooks:
- id: dont-commit-upstream
Expand Down
2 changes: 1 addition & 1 deletion cpp/include/torch_tensorrt/macros.h
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
#define STR(x) XSTR(x)

#define TORCH_TENSORRT_MAJOR_VERSION 2
#define TORCH_TENSORRT_MINOR_VERSION 6
#define TORCH_TENSORRT_MINOR_VERSION 5
#define TORCH_TENSORRT_PATCH_VERSION 0
#define TORCH_TENSORRT_VERSION \
STR(TORCH_TENSORRT_MAJOR_VERSION) \
Expand Down
19 changes: 10 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ requires = [
"typing-extensions>=4.7.0",
"future>=0.18.3",
"tensorrt-cu12==10.3.0",
"torch>=2.6.0.dev,<2.7.0",
"torch>=2.5.0.dev,<2.6.0",
"pybind11==2.6.2",
"numpy",
]
Expand Down Expand Up @@ -54,7 +54,7 @@ keywords = [
"inference",
]
dependencies = [
"torch>=2.6.0.dev,<2.7.0",
"torch>=2.5.0.dev,<2.6.0",
"tensorrt-cu12==10.3.0",
"tensorrt-cu12-bindings==10.3.0",
"tensorrt-cu12-libs==10.3.0",
Expand All @@ -65,7 +65,9 @@ dependencies = [
dynamic = ["version"]

[project.optional-dependencies]
#torchvision = ["torchvision>=0.21.dev,<0.22.0"] # Reenable once torchvision bumps to 2.6.0
torchvision = [
"torchvision",
] #Leaving torchvisions dependency unconstrained so uv can just install something that should work for the torch we have. TV's on PyT makes it hard to put version constrains in
quantization = ["nvidia-modelopt[deploy,hf,torch]~=0.17.0"]
monitoring-tools = ["rich>=13.7.1"]
jupyter = ["rich[jupyter]>=13.7.1"]
Expand All @@ -85,22 +87,21 @@ dev-dependencies = [
"pre-commit>=2.20.0",
"black>=22.6.0",
"clang-format==14.0.6",
"typos",
"mypy",
"isort",
"ruff",
"pytest",
"parameterized",
"pytest-xdist",
"parameterized>=0.2.0",
"expecttest==0.1.6",
"pyyaml",
]

environments = [
"sys_platform == 'linux'",
"sys_platform == 'windows'",
]
environments = ["sys_platform == 'linux'", "sys_platform == 'windows'"]

extra-index-url = [
"https://download.pytorch.org/whl/nightly/cu124" # We are going to define the dev enviorment as latest supported CUDA, and allow CI to handle the others, change as needed
"https://download.pytorch.org/whl/nightly/cu124", # We are going to define the dev enviorment as latest supported CUDA, and allow CI to handle the others, change as needed
]

prerelease = "if-necessary-or-explicit"
Expand Down
Loading

0 comments on commit c9a1690

Please sign in to comment.