From 395d9411db7cd2faee4a050e5400ef137fca6ae6 Mon Sep 17 00:00:00 2001 From: Angela Yi Date: Thu, 9 Jan 2025 16:52:21 -0800 Subject: [PATCH] Remove example inputs from aoti_compile_and_package Summary: The args were removed in https://github.com/pytorch/pytorch/pull/140991 Differential Revision: D67998952 --- .../torchtune/modules/_export/test_export_position_embeddings.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/torchtune/modules/_export/test_export_position_embeddings.py b/tests/torchtune/modules/_export/test_export_position_embeddings.py index 6907ca3edd..3beb23e7ef 100644 --- a/tests/torchtune/modules/_export/test_export_position_embeddings.py +++ b/tests/torchtune/modules/_export/test_export_position_embeddings.py @@ -161,7 +161,6 @@ def test_tiled_token_positional_embedding_aoti(self): with tempfile.TemporaryDirectory() as tmpdir: path = torch._inductor.aoti_compile_and_package( tpe_ep, - (self.x, self.aspect_ratio), package_path=os.path.join(tmpdir, "tpe.pt2"), ) tpe_aoti = load_package(path)