From f49d5e30eba7e45451547e007652deab93e7b191 Mon Sep 17 00:00:00 2001 From: Mikayla Gawarecki Date: Fri, 2 Aug 2024 08:51:23 -0700 Subject: [PATCH] Change owners of test/test_transformers.py to module: multi-headed-attention (#132519) So flaky tests get tagged with `module: multi-headed-attention` instead of `module: nn` Pull Request resolved: https://github.com/pytorch/pytorch/pull/132519 Approved by: https://github.com/Skylion007 --- test/test_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_transformers.py b/test/test_transformers.py index 7ac0e2da1f5d7a..af457828ebe17f 100644 --- a/test/test_transformers.py +++ b/test/test_transformers.py @@ -1,4 +1,4 @@ -# Owner(s): ["module: nn"] +# Owner(s): ["module: multi-headed-attention"] import contextlib from functools import partial