Set test owner for vmap (#67582)

Summary:
More leftover actions from https://github.com/pytorch/pytorch/issues/66232

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

Reviewed By: zou3519

Differential Revision: D32045160

Pulled By: janeyx99

fbshipit-source-id: 92ae9a533285b05b44bd04bb6127061c6fddd689
This commit is contained in:
Jane Xu 2021-11-01 07:21:09 -07:00 committed by Facebook GitHub Bot
parent 9cdd1d7e48
commit 00da7b9a3b

View File

@ -1,3 +1,5 @@
# Owner(s): ["module: vmap"]
from torch.testing._internal.common_utils import TestCase, run_tests
import torch
import torch.nn.functional as F