Skip to content

Commit 2c231da

Browse files
svekarsangelayi
andauthored
2.3 RC TEST (#2804)
* Pull 2.3 RC binaries * Turn off failing torchtext tutorials * Disable usb_semisup_learn * [export] Remove interactive aoti compilation (#2835) --------- Co-authored-by: Svetlana Karslioglu <[email protected]> Co-authored-by: Angela Yi <[email protected]>
1 parent 8e65226 commit 2c231da

File tree

4 files changed

+136
-74
lines changed

4 files changed

+136
-74
lines changed

.jenkins/build.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,8 @@ sudo apt-get install -y pandoc
2222
#Install PyTorch Nightly for test.
2323
# Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
2424
# Install 2.2 for testing - uncomment to install nightly binaries (update the version as needed).
25-
# pip uninstall -y torch torchvision torchaudio torchtext torchdata
26-
# pip3 install torch==2.2.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
27-
# pip3 install torchdata torchtext --index-url https://download.pytorch.org/whl/test/cpu
25+
pip uninstall -y torch torchvision torchaudio torchtext torchdata
26+
pip3 install torch==2.3.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
2827

2928
# Install two language tokenizers for Translation with TorchText tutorial
3029
python -m spacy download en_core_web_sm

.jenkins/metadata.json

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
"needs": "linux.16xlarge.nvidia.gpu"
3030
},
3131
"intermediate_source/torchvision_tutorial.py": {
32-
"needs": "linux.g5.4xlarge.nvidia.gpu",
32+
"needs": "linux.g5.4xlarge.nvidia.gpu",
3333
"_comment": "does not require a5g but needs to run before gpu_quantization_torchao_tutorial.py."
3434
},
3535
"advanced_source/coding_ddpg.py": {
@@ -39,6 +39,9 @@
3939
"intermediate_source/torch_compile_tutorial.py": {
4040
"needs": "linux.g5.4xlarge.nvidia.gpu"
4141
},
42+
"intermediate_source/torch_export_tutorial.py": {
43+
"needs": "linux.g5.4xlarge.nvidia.gpu"
44+
},
4245
"intermediate_source/scaled_dot_product_attention_tutorial.py": {
4346
"needs": "linux.g5.4xlarge.nvidia.gpu"
4447
},

.jenkins/validate_tutorials_built.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,15 @@
2222
"beginner_source/former_torchies/tensor_tutorial_old",
2323
"beginner_source/examples_autograd/polynomial_autograd",
2424
"beginner_source/examples_autograd/polynomial_custom_function",
25+
"beginner_source/torchtext_custom_dataset_tutorial", # not building with 2.3 RC, might be able to turn on with GA
26+
"beginner_source/text_sentiment_ngrams_tutorial", # not building with 2.3 RC, might be able to turn on with GA
2527
"beginner_source/t5_tutorial", # re-enable after this is fixed: https://github.com/pytorch/text/issues/1756
2628
"intermediate_source/mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py
2729
"intermediate_source/fx_conv_bn_fuser",
2830
"intermediate_source/_torch_export_nightly_tutorial", # does not work on release
2931
"advanced_source/super_resolution_with_onnxruntime",
3032
"advanced_source/ddp_pipeline", # requires 4 gpus
33+
"advanced_source/usb_semisup_learn", # fails with CUDA OOM error, should try on a different worker
3134
"prototype_source/fx_graph_mode_ptq_dynamic",
3235
"prototype_source/vmap_recipe",
3336
"prototype_source/torchscript_freezing",

0 commit comments

Comments
 (0)