Skip to content

Commit a002800

Browse files
committed
DLFW changes
1 parent e642f86 commit a002800

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ requires = [
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
1111
"tensorrt>=8.6,<8.7",
12-
"torch >=2.3.0.dev,<2.4.0",
12+
"torch >=2.1.0.dev,<2.4.0",
1313
"pybind11==2.6.2",
1414
"numpy",
1515
]
@@ -41,7 +41,7 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
4141
requires-python = ">=3.8"
4242
keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
4343
dependencies = [
44-
"torch >=2.3.0.dev,<2.4.0",
44+
"torch >=2.1.0.dev,<2.4.0",
4545
"tensorrt>=8.6,<8.7",
4646
"packaging>=23",
4747
"numpy",

setup.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,8 @@
3131

3232

3333
def get_root_dir() -> Path:
34-
return Path(
35-
subprocess.check_output(["git", "rev-parse", "--show-toplevel"])
36-
.decode("ascii")
37-
.strip()
38-
)
34+
dir_path = os.path.dirname(os.path.realpath(__file__))
35+
return dir_path
3936

4037

4138
def get_git_revision_short_hash() -> str:

0 commit comments

Comments
 (0)