TST Adds more modules into common module tests (#62999)
authorThomas J. Fan <thomasjpfan@gmail.com>
Wed, 25 Aug 2021 02:03:07 +0000 (19:03 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 25 Aug 2021 02:16:32 +0000 (19:16 -0700)
commitba126df61448ca3442ec77374bc32f43fcdd9773
treedf29fd0b731de63531aea051c99628d6ee821582
parent544af391b5649c8c407fa36b36631a2307997a09
TST Adds more modules into common module tests (#62999)

Summary:
This PR moves some modules into `common_modules` to see what it looks like.

While migrating some no batch modules into `common_modules`, I noticed that `desc` is not used for the name. This means we can not use `-k` to filter tests. This PR moves the sample generation into `_parametrize_test`, and passes in the already generated `module_input` into users of `modules(modules_db)`.

I can see this is a little different from opsinfo and would be happy to revert to the original implementation of `modules`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/62999

Reviewed By: heitorschueroff

Differential Revision: D30522737

Pulled By: jbschlosser

fbshipit-source-id: 7ed1aeb3753fc97a4ad6f1a3c789727c78e1bc73
torch/testing/_internal/common_modules.py