Skip to content

Commit

Permalink
Fix minimizer tests (#1006)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1006

fix tests failures introduced in D56799587

Reviewed By: hl475, kalpit-meta

Differential Revision: D57166956

fbshipit-source-id: 5c49fe28567d34f1ae2c38454f61fc1480241f2f
  • Loading branch information
zejunh authored and facebook-github-bot committed May 10, 2024
1 parent 3662d77 commit d158982
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fx2ait/fx2ait/tools/ait_minimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,12 +75,12 @@ def __init__(
self.curr_iter = 0 # We use this counter to prevent duplicate .so naming
super().__init__(module, sample_input, compare_fn, settings)

def run_a(self, mod, inputs):
def run_a(self, mod, inputs, _report_idx=-1):
mod.eval()
with torch.no_grad():
return mod(*inputs)

def run_b(self, mod, inputs):
def run_b(self, mod, inputs, _report_idx=-1):
mod.eval()
dll_name = f"{self.name}_{self.curr_iter}.so"
self.curr_iter += 1
Expand Down

0 comments on commit d158982

Please sign in to comment.