Skip to content

Commit

Permalink
Update pytorch nightly pin to 03/13
Browse files Browse the repository at this point in the history
ghstack-source-id: 449d52a683697dfc7bc5bfbf7f44751877525b11
Pull Request resolved: #2409
  • Loading branch information
SS-JIA committed Mar 13, 2024
1 parent ad9f186 commit d2c2043
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .ci/docker/ci_commit_pins/pytorch.txt
Original file line number Diff line number Diff line change
@@ -1 +1 @@
6ca9ae4f8693639c395544327f7e362441a58c79
9f235971f02e0d53038f5bcef9b7018be2ac8c6d
2 changes: 1 addition & 1 deletion install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ fi
# models in executorch/examples/models.
# The version in this file will be the correct version for the
# corresponsing version of the repo.
NIGHTLY_VERSION=dev20240312
NIGHTLY_VERSION=dev20240313

TORCH_VERSION=2.3.0.${NIGHTLY_VERSION}
pip install --force-reinstall --pre torch=="${TORCH_VERSION}" -i https://download.pytorch.org/whl/nightly/cpu
Expand Down
2 changes: 1 addition & 1 deletion third-party/pytorch
Submodule pytorch updated 87 files
+14 −10 .ci/docker/common/install_conda.sh
+1 −1 .github/ci_commit_pins/xla.txt
+5 −5 Dockerfile
+13 −13 aten/src/ATen/native/DispatchStub.h
+28 −1 aten/src/ATen/native/vulkan/api/Context.h
+8 −0 aten/src/ATen/native/vulkan/api/Resource.h
+3 −0 aten/src/ATen/native/vulkan/api/ShaderRegistry.h
+140 −15 aten/src/ATen/native/vulkan/api/Tensor.cpp
+75 −4 aten/src/ATen/native/vulkan/api/Tensor.h
+15 −2 aten/src/ATen/native/vulkan/api/Utils.h
+31 −0 benchmarks/dynamo/common.py
+31 −17 c10/util/Logging.cpp
+6 −0 caffe2/CMakeLists.txt
+4 −4 caffe2/python/examples/imagenet_trainer.py
+2 −2 caffe2/python/hypothesis_test.py
+11 −12 caffe2/python/operator_test/rank_loss_operator_test.py
+1 −1 caffe2/python/utils.py
+2 −12 docs/source/distributed.checkpoint.rst
+2 −1 docs/source/torch.compiler_aot_inductor.rst
+13 −8 scripts/release/apply-release-changes.sh
+24 −0 scripts/release/tag-docker-images.sh
+19 −0 test/distributed/_tensor/test_redistribute.py
+3 −27 test/distributed/checkpoint/e2e/test_e2e_save_and_load.py
+1 −1 test/distributed/checkpoint/test_fsspec.py
+23 −24 test/distributed/checkpoint/test_state_dict.py
+1 −73 test/distributed/tensor/parallel/test_tp_examples.py
+2 −2 test/distributed/test_c10d_nccl.py
+2 −2 test/dynamo/test_guard_manager.py
+0 −195 test/dynamo/test_replay_record.py
+9 −4 test/export/test_export.py
+2 −1 test/export/test_export_nonstrict.py
+23 −1 test/export/test_serdes.py
+102 −92 test/export/test_serialize.py
+25 −2 test/export/test_torchbind.py
+5 −0 test/export/testing.py
+53 −0 test/fx/test_partitioner_order.py
+358 −0 test/inductor/test_cutlass_backend.py
+0 −312 test/inductor/test_max_autotune.py
+0 −142 test/inductor/test_pad_mm.py
+111 −9 test/inductor/test_triton_kernels.py
+47 −0 test/quantization/pt2e/test_quantize_pt2e.py
+2 −1 test/run_test.py
+5 −3 test/test_dynamic_shapes.py
+2 −0 test/test_sparse.py
+4 −2 test/test_sparse_csr.py
+67 −1 tools/gen_vulkan_spv.py
+1 −0 torch/_C/_autograd.pyi
+8 −6 torch/_dynamo/convert_frame.py
+7 −0 torch/_dynamo/guards.py
+42 −2 torch/_dynamo/output_graph.py
+20 −8 torch/_dynamo/utils.py
+7 −0 torch/_dynamo/variables/builder.py
+1 −1 torch/_dynamo/variables/distributed.py
+9 −3 torch/_functorch/_aot_autograd/jit_compile_runtime_wrappers.py
+48 −33 torch/_higher_order_ops/triton_kernel_wrap.py
+11 −0 torch/_inductor/codegen/wrapper.py
+2 −64 torch/_inductor/fx_passes/pad_mm.py
+4 −4 torch/_inductor/ir.py
+7 −1 torch/_ops.py
+2 −5 torch/_subclasses/fake_tensor.py
+114 −31 torch/ao/quantization/fx/_decomposed.py
+7 −0 torch/ao/quantization/quantizer/composable_quantizer.py
+1 −0 torch/autograd/profiler.py
+4 −0 torch/autograd/profiler_util.py
+1 −1 torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp
+5 −0 torch/csrc/dynamo/cache_entry.cpp
+2 −0 torch/csrc/dynamo/cache_entry.h
+10 −3 torch/csrc/dynamo/extra_state.cpp
+20 −37 torch/csrc/dynamo/guards.cpp
+8 −0 torch/csrc/dynamo/guards.h
+0 −4 torch/distributed/_tensor/api.py
+13 −8 torch/distributed/_tensor/redistribute.py
+0 −1 torch/distributed/checkpoint/__init__.py
+100 −0 torch/distributed/checkpoint/_checkpointer.py
+116 −9 torch/distributed/checkpoint/_fsspec_filesystem.py
+1 −1 torch/distributed/checkpoint/_storage_utils.py
+0 −122 torch/distributed/checkpoint/fsspec.py
+0 −7 torch/distributed/checkpoint/state_dict.py
+4 −2 torch/distributed/tensor/parallel/style.py
+2 −1 torch/export/_safeguard.py
+8 −2 torch/export/_trace.py
+23 −12 torch/fx/experimental/symbolic_shapes.py
+8 −0 torch/fx/graph.py
+7 −6 torch/fx/passes/infra/partitioner.py
+10 −7 torch/fx/proxy.py
+119 −30 torch/testing/_internal/distributed/_tensor/common_dtensor.py
+22 −2 torch/utils/_triton.py

0 comments on commit d2c2043

Please sign in to comment.