Merge branch 'main' into deprecate-torchtext-tutorials

Author: svekars

.jenkins/build.sh

@@ -21,9 +21,10 @@ sudo apt-get install -y pandoc
 
 #Install PyTorch Nightly for test.
 # Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
-# Install 2.4 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
-# pip uninstall -y torch torchvision torchaudio torchtext torchdata
-# pip3 install torch==2.4.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
+# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
+pip uninstall -y torch torchvision torchaudio torchtext torchdata
+pip3 install torch==2.5.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
+pip3 install fbgemm-gpu==1.0.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
 
 # Install two language tokenizers for Translation with TorchText tutorial
 python -m spacy download en_core_web_sm

recipes_source/regional_compilation.py

@@ -3,6 +3,7 @@
 ============================================================================
 
 **Author:** `Animesh Jain <https://github.com/anijain2305>`_
+
 As deep learning models get larger, the compilation time of these models also
 increases. This extended compilation time can result in a large startup time in
 inference services or wasted resources in large-scale training. This recipe
@@ -23,7 +24,7 @@
 
    pip install torch
 
-.. note:: 
+.. note::
    This feature is available starting with the 2.5 release. If you are using version 2.4,
    you can enable the configuration flag ``torch._dynamo.config.inline_inbuilt_nn_modules=True``
    to prevent recompilations during regional compilation. In version 2.5, this flag is enabled by default.