Skip to content

Commit 8496f27

Browse files
authored
Update torch pin to Sep 06, 2025 (#14129)
Summary: Branch cut for PT/PT is Sep 08, so let's bring it forward just a few days before the branch cut. Once we cut the 1.0 branch cut in ET, we will pin to 2.9 which is basically equivalent of Sep 08
1 parent dc944fe commit 8496f27

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

.ci/docker/ci_commit_pins/pytorch.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
e7152ff8a6a929a0db7f3f4a72a5b6d471769cd3
1+
4d4abec80f03cd8fdefe1d9cb3a60d3690cd777e

install_requirements.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def python_is_compatible():
7171
#
7272
# NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
7373
# by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
74-
NIGHTLY_VERSION = "dev20250811"
74+
NIGHTLY_VERSION = "dev20250906"
7575

7676

7777
def install_requirements(use_pytorch_nightly):

runtime/core/portable_type/c10/torch/headeronly/macros/Macros.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,8 @@ using namespace c10::xpu;
259259
// to resolve potential warnings.
260260
#if __CUDA_ARCH__ == 750
261261
constexpr uint32_t CUDA_MAX_THREADS_PER_SM = 1024;
262-
#elif __CUDA_ARCH__ == 860 || __CUDA_ARCH__ == 870 || __CUDA_ARCH__ == 890
262+
#elif __CUDA_ARCH__ == 860 || __CUDA_ARCH__ == 870 || __CUDA_ARCH__ == 890 || \
263+
__CUDA_ARCH__ == 1200
263264
constexpr uint32_t CUDA_MAX_THREADS_PER_SM = 1536;
264265
#else
265266
constexpr uint32_t CUDA_MAX_THREADS_PER_SM = 2048;

0 commit comments

Comments
 (0)