Skip to content

Commit f050f1a

Browse files
DiweiSunchunyuan-w
andauthored
Update dependency_version.json (#3756)
* Update dependency_version.json * Update dependency_version.json * change guard_int back to evaluate_static_shape on release/2.8 --------- Co-authored-by: y <[email protected]>
1 parent cefe583 commit f050f1a

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

dependency_version.json

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,14 @@
44
"min-version": "12.3.0"
55
},
66
"pytorch": {
7-
"index-url": "https://download.pytorch.org/whl/nightly/cpu",
8-
"version": "2.8.0.dev20250626+cpu"
7+
"index-url": "https://download.pytorch.org/whl/test/cpu",
8+
"version": "2.8.0+cpu"
99
},
1010
"torchaudio": {
11-
"version": "2.8.0.dev20250626+cpu"
11+
"version": "2.8.0+cpu"
1212
},
1313
"torchvision": {
14-
"version": "0.23.0.dev20250626+cpu"
14+
"version": "0.23.0+cpu"
1515
},
1616
"deepspeed": {
1717
"version": "0.17.1",

tests/cpu/test_inductor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -234,9 +234,9 @@ def _mm(a: TensorBox, b: TensorBox):
234234
reduced_sizes = [k]
235235
new_size = [m, n]
236236

237-
m = V.graph.sizevars.guard_int(m)
238-
n = V.graph.sizevars.guard_int(n)
239-
k = V.graph.sizevars.guard_int(k)
237+
m = V.graph.sizevars.evaluate_static_shape(m)
238+
n = V.graph.sizevars.evaluate_static_shape(n)
239+
k = V.graph.sizevars.evaluate_static_shape(k)
240240

241241
_a_loader = a.make_loader()
242242
_b_loader = b.make_loader()

0 commit comments

Comments
 (0)