Skip to content

Commit 95d0c24

Browse files
authored
Bump up torch nightly version (#437)
This commit bumps up torch nightly version. It includes some changes to resolve errors caused by bump-up. - Remove PT2E Qunatization. - torch has introduced a new library "torchao" that includes pt2e related codes. - pt2e quantization was introduced to TICO for test purposes. - Which means it doesn't have users. - Keep tracking the new library is too burden to have pt2e features in TICO. - Replace PT2E related codes with the codes using `wrapq`. - Suppress future warnings caused by torch. - minor changes that are needed according to the bump-ups. TICO-DCO-1.0-Signed-off-by: seongwoo <mhs4670go@naver.com>
1 parent e49ffdb commit 95d0c24

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

44 files changed

+176
-2142
lines changed

infra/dependency/torch_dev.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
torch==2.10.0.dev20251012+cpu
1+
torch==2.11.0.dev20260113+cpu
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
torchvision==0.25.0.dev20251012+cpu
1+
torchvision==0.25.0.dev20260113+cpu

test/pt2_to_circle_test/test_pt2_to_circle.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,8 @@ def convert_pt2_to_circle(
9393
circle_model_path: str,
9494
config: Optional[CompileConfigBase] = None,
9595
):
96-
tico.pt2_to_circle.convert(pt2_model_path, circle_model_path, config=config)
96+
with SuppressWarning(FutureWarning, ".*LeafSpec*"):
97+
tico.pt2_to_circle.convert(pt2_model_path, circle_model_path, config=config)
9798

9899

99100
@print_name_on_exception
@@ -112,7 +113,8 @@ def convert_nnmodule_to_circle(
112113
kwargs=forward_kwargs,
113114
dynamic_shapes=dynamic_shapes,
114115
)
115-
circle_program = convert_exported_module_to_circle(exported_program, config)
116+
with SuppressWarning(FutureWarning, ".*LeafSpec*"):
117+
circle_program = convert_exported_module_to_circle(exported_program, config)
116118
circle_binary = circle_program
117119
with open(circle_model_path, "wb") as f:
118120
f.write(circle_binary)

test/pt2_to_qcircle_test/__init__.py

Lines changed: 0 additions & 30 deletions
This file was deleted.

test/pt2_to_qcircle_test/builder.py

Lines changed: 0 additions & 122 deletions
This file was deleted.

test/pt2_to_qcircle_test/test_op.py

Lines changed: 0 additions & 21 deletions
This file was deleted.

0 commit comments

Comments
 (0)