3 Commits

Author	SHA1	Message	Date
Anthony Barbier	c8d44a2296	Add __main__ guards to fx tests (#154715) ... This PR is part of a series attempting to re-submit #134592 as smaller PRs. In fx tests: - Add and use a common raise_on_run_directly method for when a user runs a test file directly which should not be run this way. Print the file which the user should have run. - Raise a RuntimeError on tests which have been disabled (not run) - Remove any remaining uses of "unittest.main()"" Pull Request resolved: https://github.com/pytorch/pytorch/pull/154715 Approved by: https://github.com/Skylion007	2025-06-04 14:38:50 +00:00
Shangdi Yu	51cbac4e6a	[export] Change fx graph _replace_hook to a list of Callable (#142006) ... Summary: Change fx graph module's _replace_hook from a single hook, to a list of hooks. This is to prepare to registering more hooks for inductor provenance tracking, where we might need to register multiple hooks for node replacement. Test Plan: ``` buck run mode/dev-nosan caffe2/test:fx -- -r test_hooks_for_node_update buck run mode/dev-nosan caffe2/test:test_export -- -r test_replace_hook ``` Differential Revision: D66726724 Pull Request resolved: https://github.com/pytorch/pytorch/pull/142006 Approved by: https://github.com/zhxchen17	2024-12-05 03:26:48 +00:00
Sheng Fu	bbeb0906c4	Register creak_node_hook (#126671) ... Differential Revision: D57469157 Pull Request resolved: https://github.com/pytorch/pytorch/pull/126671 Approved by: https://github.com/angelayi	2024-05-24 23:32:15 +00:00