Skip to content

Commit 3d84a64

Browse files
AlexanderDokuchaevAlexanderDokuchaev
committedMar 21, 2025
test alignment
1 parent dd0ef13 commit 3d84a64

13 files changed

+1119
-8
lines changed
 

‎nncf/torch/model_creation.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -355,10 +355,10 @@ def wrap_model(
355355
if not trace_parameters:
356356
msg = "The 'trace_parameters=False' option is not supported in the experimental tracing mode."
357357
raise nncf.InternalError(msg)
358-
from nncf.experimental.torch2.function_hook import wrap_model
358+
from nncf.experimental.torch2.function_hook import wrap_model as pt2_wrap_model
359359
from nncf.experimental.torch2.function_hook.nncf_graph.nncf_graph_builder import GraphModelWrapper
360360

361-
wrapped_model = GraphModelWrapper(wrap_model(model), example_input=example_input)
361+
wrapped_model = GraphModelWrapper(pt2_wrap_model(model), example_input=example_input)
362362
return wrapped_model
363363

364364
if not isinstance(model, torch.nn.Module):

‎tests/torch2/data/reference_scales/TwoConvTestModel_overflow_fix_disable.json

+72
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,72 @@
1+
{
2+
"__nncf_hooks.pre_hooks.features/0/0/conv2d/0__1.0": {
3+
"input_low": [
4+
[
5+
[
6+
[
7+
-1.0
8+
]
9+
]
10+
],
11+
[
12+
[
13+
[
14+
-1.0
15+
]
16+
]
17+
]
18+
],
19+
"input_high": [
20+
[
21+
[
22+
[
23+
1.0
24+
]
25+
]
26+
],
27+
[
28+
[
29+
[
30+
1.0
31+
]
32+
]
33+
]
34+
]
35+
},
36+
"__nncf_hooks.pre_hooks.features/1/0/conv2d/0__1.0": {
37+
"input_low": [
38+
[
39+
[
40+
[
41+
-1.0
42+
]
43+
]
44+
]
45+
],
46+
"input_high": [
47+
[
48+
[
49+
[
50+
1.0
51+
]
52+
]
53+
]
54+
]
55+
},
56+
"__nncf_hooks.post_hooks.x__0.0": {
57+
"input_low": [
58+
0.0
59+
],
60+
"input_high": [
61+
0.9970665574073792
62+
]
63+
},
64+
"__nncf_hooks.post_hooks.features/0/0/conv2d/0__0.0": {
65+
"input_low": [
66+
-3.8243322372436523
67+
],
68+
"input_high": [
69+
3.794454574584961
70+
]
71+
}
72+
}

‎tests/torch2/data/reference_scales/TwoConvTestModel_overflow_fix_enable.json

+72
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,72 @@
1+
{
2+
"__nncf_hooks.pre_hooks.features/0/0/conv2d/0__1.0": {
3+
"input_low": [
4+
[
5+
[
6+
[
7+
-2.0
8+
]
9+
]
10+
],
11+
[
12+
[
13+
[
14+
-2.0
15+
]
16+
]
17+
]
18+
],
19+
"input_high": [
20+
[
21+
[
22+
[
23+
2.0
24+
]
25+
]
26+
],
27+
[
28+
[
29+
[
30+
2.0
31+
]
32+
]
33+
]
34+
]
35+
},
36+
"__nncf_hooks.pre_hooks.features/1/0/conv2d/0__1.0": {
37+
"input_low": [
38+
[
39+
[
40+
[
41+
-2.0
42+
]
43+
]
44+
]
45+
],
46+
"input_high": [
47+
[
48+
[
49+
[
50+
2.0
51+
]
52+
]
53+
]
54+
]
55+
},
56+
"__nncf_hooks.post_hooks.x__0.0": {
57+
"input_low": [
58+
0.0
59+
],
60+
"input_high": [
61+
0.9970665574073792
62+
]
63+
},
64+
"__nncf_hooks.post_hooks.features/0/0/conv2d/0__0.0": {
65+
"input_low": [
66+
-3.8243322372436523
67+
],
68+
"input_high": [
69+
3.794454574584961
70+
]
71+
}
72+
}

‎tests/torch2/data/reference_scales/TwoConvTestModel_overflow_fix_first_layer_only.json

+72
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,72 @@
1+
{
2+
"__nncf_hooks.pre_hooks.features/0/0/conv2d/0__1.0": {
3+
"input_low": [
4+
[
5+
[
6+
[
7+
-2.0
8+
]
9+
]
10+
],
11+
[
12+
[
13+
[
14+
-2.0
15+
]
16+
]
17+
]
18+
],
19+
"input_high": [
20+
[
21+
[
22+
[
23+
2.0
24+
]
25+
]
26+
],
27+
[
28+
[
29+
[
30+
2.0
31+
]
32+
]
33+
]
34+
]
35+
},
36+
"__nncf_hooks.pre_hooks.features/1/0/conv2d/0__1.0": {
37+
"input_low": [
38+
[
39+
[
40+
[
41+
-1.0
42+
]
43+
]
44+
]
45+
],
46+
"input_high": [
47+
[
48+
[
49+
[
50+
1.0
51+
]
52+
]
53+
]
54+
]
55+
},
56+
"__nncf_hooks.post_hooks.x__0.0": {
57+
"input_low": [
58+
0.0
59+
],
60+
"input_high": [
61+
0.9970665574073792
62+
]
63+
},
64+
"__nncf_hooks.post_hooks.features/0/0/conv2d/0__0.0": {
65+
"input_low": [
66+
-3.8243322372436523
67+
],
68+
"input_high": [
69+
3.794454574584961
70+
]
71+
}
72+
}

0 commit comments

Comments
 (0)
Failed to load comments.