Skip to content

Commit

Permalink
[BE] Beef up test_functionalization to test functionalizing multi-par…
Browse files Browse the repository at this point in the history
…ameter functions (pytorch#89798)

Previously, `assert_functionalization` only took in uni-Tensor-parameter functions. This PR beefs up the check to allow for functions that take multiple parameters.

This PR also changes the test_instance_norm test to check that the multiparam change works.

## Test plan
Locally tested, CI should also pass.
Pull Request resolved: pytorch#89798
Approved by: https://github.com/samdow
  • Loading branch information
janeyx99 authored and pytorchmergebot committed Nov 30, 2022
1 parent 4144ad1 commit 76e869c
Showing 1 changed file with 203 additions and 269 deletions.
Loading

0 comments on commit 76e869c

Please sign in to comment.