Skip to content

Commit 136c1d0

Browse files
authored
fix test_optimize_linear_bn_parameters_behavior and update dependency_version.json (#3401)
1 parent 1e1bd6b commit 136c1d0

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

dependency_version.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77
"commit": "llvmorg-16.0.6"
88
},
99
"pytorch": {
10-
"version": "2.6.0.dev20241124+cpu"
10+
"version": "2.6.0.dev20241211+cpu"
1111
},
1212
"torchaudio": {
13-
"version": "2.5.0.dev20241121+cpu"
13+
"version": "2.5.0.dev20241211+cpu"
1414
},
1515
"torchvision": {
16-
"version": "0.20.0.dev20241121+cpu"
16+
"version": "0.20.0.dev20241211+cpu"
1717
},
1818
"torch-ccl": {
1919
"commit": "torch_ccl_dev_2.6",

tests/cpu/test_ipex_optimize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def test_optimize_linear_bn_parameters_behavior(self):
165165
any(
166166
n.kind() == "ipex::batch_norm" for n in trace_graph.nodes()
167167
),
168-
not (linear_bn_folding),
168+
not (linear_bn_folding or (dim == 1 and level == "O1")),
169169
)
170170

171171
def test_optimize_conv_bn_linear_bn_parameters_behavior(self):

0 commit comments

Comments
 (0)