github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Enable Torch-Inductor Part 2

Browse Source
This commit is contained in:
mariecwhite
2023-01-10 17:43:26 -08:00
parent 72f29b67d5
commit d0b613d22e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
shark/shark_benchmark_runner.py
View File

@@ -118,8 +118,8 @@ class SharkBenchmarkRunner(SharkRunner):
"cuda:0" if self.device == "cuda" else "cpu"
)
HFmodel, input = get_torch_model(modelname)[:2]
HFmodel = dynamo.optimize("inductor")(HFmodel)
frontend_model = HFmodel.model
frontend_model = dynamo.optimize("inductor")(frontend_model)
frontend_model.to(torch_device)
input.to(torch_device)