PR #18547 Fix memory leak issue in `torch_fx` tests

Fix memory leak issue in `torch_fx` tests #18547

ydshieh merged 5 commits into huggingface:main from ydshieh:fix_torch_fx_test_mem_issue

ydshieh commented on 2022-08-10

ydshieh requested a review from

LysandreJik 3 years ago

ydshieh requested a review from

michaelbenayoun 3 years ago

ydshieh removed review request from

michaelbenayoun 3 years ago

ydshieh requested a review from

sgugger 3 years ago

ydshieh commented on 2022-08-10

sgugger commented on 2022-08-10

ydshieh requested a review from

michaelbenayoun 3 years ago

michaelbenayoun commented on 2022-08-11

ydshieh force pushed 3 years ago

michaelbenayoun approved these changes on 2022-08-22

ydshieh force pushed 3 years ago

LysandreJik approved these changes on 2022-08-24

try to run torch_fx in a spawn process to avoid memory issue

656fd024

fix quality

0902044e

Update tests/test_modeling_common.py

f0973347

Add comment

35399319

simply call clear_torch_jit_class_registry

7d03df46

ydshieh force pushed to 7d03df46 3 years ago

ydshieh changed the title ~~Run torch_fx tests in a spawn process to avoid memory issue~~ Fix memory leak issue in `torch_fx` tests 3 years ago

ydshieh merged 8b67f209 into main 3 years ago

ydshieh deleted the fix_torch_fx_test_mem_issue branch 3 years ago

Reviewers

LysandreJik

michaelbenayoun

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone