github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-10 06:17:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Revert changes to multiprocessing (#585)

Browse Source
This commit is contained in:
Ean Garvey
2022-12-07 19:59:17 -06:00
committed by GitHub
parent 40eea21863
commit 9a14298146
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tank/test_models.py
View File

@@ -230,21 +230,13 @@ class SharkModuleTester:
def run_test(module_tester, dynamic, device):
import multiprocessing
tempdir = tempfile.TemporaryDirectory(
prefix=module_tester.tmp_prefix, dir="./shark_tmp/"
)
module_tester.temp_dir = tempdir.name
with ireec.tools.TempFileSaver(tempdir.name):
p = multiprocessing.Process(
target=module_tester.create_and_check_module,
args=(dynamic, device),
)
p.start()
p.join()
return p
module_tester.create_and_check_module(dynamic, device)
class SharkModuleTest(unittest.TestCase):
@@ -434,5 +426,8 @@ class SharkModuleTest(unittest.TestCase):
# We must create a new process each time we benchmark a model to allow
# for Tensorflow to release GPU resources. Using the same process to
# benchmark multiple models leads to OOM.
run_test(self.module_tester, dynamic, device)
p = multiprocessing.Process(
target=run_test, args=(self.module_tester, dynamic, device)
)
p.start()
p.join()