Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
104 changes: 0 additions & 104 deletions .github/workflows/blossom-ci.yml

This file was deleted.

3 changes: 3 additions & 0 deletions .github/workflows/build-test-linux-aarch64-jetpack.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-test-linux-aarch64.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-test-linux-x86_64.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-test-linux-x86_64_rtx.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-test-windows.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-test-windows_rtx.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@ on:
- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
workflow_dispatch:

permissions:
contents: read

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
Expand Down
4 changes: 1 addition & 3 deletions .github/workflows/codeql.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,7 @@ on:
branches: [ "main" ]
pull_request:
branches: [ "main" ]
schedule:
- cron: '26 11 * * 2'
workflow_dispatch:
workflow_dispatch:

jobs:
analyze:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/docgen.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,9 @@ on:
push:
branches: [main]

permissions:
contents: read

jobs:
build-docs:
runs-on: linux.g5.4xlarge.nvidia.gpu
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/docker_builder.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,9 @@ on:
- 'docs/**'
workflow_dispatch:

permissions:
contents: read

# If pushes to main are made in rapid succession,
# cancel existing docker builds and use newer commits
concurrency:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/executorch-static-linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,9 @@ on:
default: ""
type: string

permissions:
contents: read

jobs:
select-matrix:
runs-on: ubuntu-latest
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/label.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,9 @@
name: Labeler
on: [pull_request_target]

permissions:
contents: read

jobs:
label:
permissions:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/linter.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@ on:
pull_request:
types: [opened, synchronize, ready_for_review, review_requested, reopened]

permissions:
contents: read

jobs:
cpp-linting:
name: C++ Linting
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/linux-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,9 @@ on:
type: boolean
required: false

permissions:
contents: read

jobs:
test:
strategy:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/nightlies.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,9 @@ on:
- cron: '30 9 * * *'
workflow_dispatch:

permissions:
contents: read

jobs:
cut_nightly:
runs-on: ubuntu-latest
Expand Down
5 changes: 5 additions & 0 deletions .github/workflows/stale.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ on:
schedule:
- cron: "0 0 * * *"

permissions:
contents: write
issues: write
pull-requests: write

jobs:
stale:

Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/windows-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,9 @@ on:
description: "Whether to use TensorRT-RTX"
default: false
type: boolean
permissions:
contents: read

jobs:
test:
strategy:
Expand Down
41 changes: 24 additions & 17 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -339,25 +339,32 @@ def copy_libtorchtrt(multilinux=False, rt_only=False):
dir_path + "/trtorch/lib/libtrtorch.so",
)
elif rt_only:
os.system(
"tar -xzf "
+ dir_path
+ "/../bazel-bin/libtorchtrt_runtime.tar.gz --strip-components=1 -C "
+ dir_path
+ "/torch_tensorrt"
subprocess.run(
[
"tar",
"-xzf",
dir_path + "/../bazel-bin/libtorchtrt_runtime.tar.gz",
"--strip-components=1",
"-C",
dir_path + "/torch_tensorrt",
],
check=True,
)
else:
os.system(
"tar -xzf "
+ dir_path
+ "/../bazel-bin/libtorchtrt.tar.gz "
+ "--exclude='torch_tensorrt/src' "
+ "--exclude='torch_tensorrt/src/*' "
+ "--exclude='torch_tensorrt/examples' "
+ "--exclude='torch_tensorrt/examples/*' "
+ "--strip-components=1 -C "
+ dir_path
+ "/torch_tensorrt"
subprocess.run(
[
"tar",
"-xzf",
dir_path + "/../bazel-bin/libtorchtrt.tar.gz",
"--exclude=torch_tensorrt/src",
"--exclude=torch_tensorrt/src/*",
"--exclude=torch_tensorrt/examples",
"--exclude=torch_tensorrt/examples/*",
"--strip-components=1",
"-C",
dir_path + "/torch_tensorrt",
],
check=True,
)


Expand Down
3 changes: 2 additions & 1 deletion tests/modules/hub.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,8 @@ def main():
manifest = {"version": torch_version}

# Creating an empty manifest file for overwriting post setup
os.system("touch {}".format(MANIFEST_FILE))
with open(MANIFEST_FILE, "a"):
pass
else:
manifest_exists = True

Expand Down
3 changes: 2 additions & 1 deletion tools/perf/hub.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,8 @@ def main():
manifest = {"version": torch_version}

# Creating an empty manifest file for overwriting post setup
os.system("touch {}".format(MANIFEST_FILE))
with open(MANIFEST_FILE, "a"):
pass
else:
# Load manifest if already exists
with open(MANIFEST_FILE, "r") as f:
Expand Down
Loading