Skip to content

Commit ea8de08

Browse files
narendasanperi044
narendasan
and
peri044
authored
chore: Move master to test nightly only (#1370)
* chore: Fix centralized partititoning Signed-off-by: Dheeraj Peri <[email protected]> * chore: Add new parititoning headers to py packaging Signed-off-by: Dheeraj Peri <[email protected]> * chore: Add partitioning headers to C++ tar package Signed-off-by: Dheeraj Peri <[email protected]> * chore: Move master to test nightly only Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]> * chore: Different build Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]> * Updated config.yml * Updated config.yml * Update CI conf Signed-off-by: Naren Dasan <[email protected]> * chore: reordering dep installation Signed-off-by: Naren Dasan <[email protected]> * chore: Allow prerelease deps Signed-off-by: Naren Dasan <[email protected]> * chore: relax torch version Signed-off-by: Naren Dasan <[email protected]> * chore: update cudnn for pytorch Signed-off-by: Naren Dasan <[email protected]> * chore: switch back to cuda 11.6 because tensorrt is pinned to cudnn 8.4 Signed-off-by: Naren Dasan <[email protected]> * chore: release configs Signed-off-by: Naren Dasan <[email protected]> * chore: update build container Signed-off-by: Naren Dasan <[email protected]> * update ci release container Signed-off-by: Naren Dasan <[email protected]> * Downgrade default libtorch to cuda 11.6 due to torchvision / tensorrt incompatibility Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Dheeraj Peri <[email protected]> Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]> Co-authored-by: Dheeraj Peri <[email protected]>
1 parent bf2054e commit ea8de08

File tree

9 files changed

+148
-310
lines changed

9 files changed

+148
-310
lines changed

‎.circleci/config.yml

Lines changed: 129 additions & 292 deletions
Large diffs are not rendered by default.

‎WORKSPACE

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,17 +56,17 @@ new_local_repository(
5656
http_archive(
5757
name = "libtorch",
5858
build_file = "@//third_party/libtorch:BUILD",
59-
sha256 = "5a392132fbff9db1482eae72a30f74b09f53a47edf8305fe9688d4ce7ddb0b6b",
59+
sha256 = "486106ddc5b5ad532f030f447940a571b924da821b9534d25c0cef5503cdfaea",
6060
strip_prefix = "libtorch",
61-
urls = ["https://download.pytorch.org/libtorch/cu116/libtorch-cxx11-abi-shared-with-deps-1.12.1%2Bcu116.zip"],
61+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-cxx11-abi-shared-with-deps-1.13.0.dev20220921%2Bcu116.zip"],
6262
)
6363

6464
http_archive(
6565
name = "libtorch_pre_cxx11_abi",
6666
build_file = "@//third_party/libtorch:BUILD",
67-
sha256 = "5e044cc56a29cd4f3a7198c0fe5b2f0fa8f4c38cd71a0491274b6a914e8f24a7",
67+
sha256 = "b304ebf26effcbbefcec99134bcfb0127c499306343fbe2e2cd127213448a4a6",
6868
strip_prefix = "libtorch",
69-
urls = ["https://download.pytorch.org/libtorch/cu116/libtorch-shared-with-deps-1.12.1%2Bcu116.zip"],
69+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-shared-with-deps-1.13.0.dev20220921%2Bcu116.zip"],
7070
)
7171

7272
# Download these tarballs manually from the NVIDIA website

‎py/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM pytorch/manylinux-builder:cuda11.3
1+
FROM pytorch/manylinux-builder:cuda11.6
22

33
RUN yum install -y ninja-build
44

‎py/ci/Dockerfile.ci

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM pytorch/manylinux-builder:cuda11.3
1+
FROM pytorch/manylinux-builder:cuda11.6
22
ARG trt_version
33

44
RUN echo -e "Installing with TensorRT ${trt_version}"

‎py/ci/build_whl.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
cd /workspace/project/py
66

77
export CXX=g++
8-
export CUDA_HOME=/usr/local/cuda-11.3
8+
export CUDA_HOME=/usr/local/cuda-11.6
99
PROJECT_DIR=/workspace/project
1010

1111
cp -r $CUDA_HOME /usr/local/cuda

‎py/requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
numpy
22
pybind11==2.6.2
33
--extra-index-url https://download.pytorch.org/whl/nightly/cu116
4-
torch==1.13.0.dev20220825+cu116
4+
torch==1.13.0.dev20220921+cu116
5+
torchvision==0.14.0.dev20220921+cu116
56
--extra-index-url https://pypi.ngc.nvidia.com
67
nvidia-tensorrt==8.4.3.1

‎py/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,7 @@ def run(self):
376376
long_description=long_description,
377377
ext_modules=ext_modules,
378378
install_requires=[
379-
"torch>=1.12.0+cu113,<1.13.0",
379+
"torch>=1.13.0.dev0,<1.14.0",
380380
],
381381
setup_requires=[],
382382
cmdclass={

‎toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ local_repository(
4141
new_local_repository(
4242
name = "cuda",
4343
build_file = "@//third_party/cuda:BUILD",
44-
path = "/usr/local/cuda-11.3",
44+
path = "/usr/local/cuda-11.6",
4545
)
4646

4747
new_local_repository(
@@ -56,17 +56,17 @@ new_local_repository(
5656
http_archive(
5757
name = "libtorch",
5858
build_file = "@//third_party/libtorch:BUILD",
59-
sha256 = "80f089939de20e68e3fcad4dfa72a26c8bf91b5e77b11042f671f39ebac35865",
6059
strip_prefix = "libtorch",
61-
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.12.0%2Bcu113.zip"],
60+
sha256 = "94573e4c5f84ee70dea00c39dc1afcd4173ef4454f4d7cb428b499cf4c3b86e1",
61+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-cxx11-abi-shared-with-deps-1.13.0.dev20220922%2Bcu116.zip"],
6262
)
6363

6464
http_archive(
6565
name = "libtorch_pre_cxx11_abi",
6666
build_file = "@//third_party/libtorch:BUILD",
67-
sha256 = "8e35371403f7052d9e9b43bcff383980dbde4df028986dc1dab539953481d55f",
6867
strip_prefix = "libtorch",
69-
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.12.0%2Bcu113.zip"],
68+
sha256 = "5748ee06d81934a4bf7654653b34820365b264d16f6b99b56833cfdd3005da71",
69+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-shared-with-deps-1.13.0.dev20220922%2Bcu116.zip"],
7070
)
7171

7272
####################################################################################

‎toolchains/ci_workspaces/WORKSPACE.x86_64.release.ubuntu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,17 +56,17 @@ new_local_repository(
5656
http_archive(
5757
name = "libtorch",
5858
build_file = "@//third_party/libtorch:BUILD",
59-
sha256 = "80f089939de20e68e3fcad4dfa72a26c8bf91b5e77b11042f671f39ebac35865",
6059
strip_prefix = "libtorch",
61-
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.12.0%2Bcu113.zip"],
60+
sha256 = "94573e4c5f84ee70dea00c39dc1afcd4173ef4454f4d7cb428b499cf4c3b86e1",
61+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-cxx11-abi-shared-with-deps-1.13.0.dev20220922%2Bcu116.zip"],
6262
)
6363

6464
http_archive(
6565
name = "libtorch_pre_cxx11_abi",
6666
build_file = "@//third_party/libtorch:BUILD",
67-
sha256 = "8e35371403f7052d9e9b43bcff383980dbde4df028986dc1dab539953481d55f",
6867
strip_prefix = "libtorch",
69-
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.12.0%2Bcu113.zip"],
68+
sha256 = "5748ee06d81934a4bf7654653b34820365b264d16f6b99b56833cfdd3005da71",
69+
urls = ["https://download.pytorch.org/libtorch/nightly/cu116/libtorch-shared-with-deps-1.13.0.dev20220922%2Bcu116.zip"],
7070
)
7171

7272
####################################################################################

0 commit comments

Comments
 (0)