diff --git a/test/test_kernel_launch_checks.py b/test/test_kernel_launch_checks.py index 10479e7d13b1..278026a021d7 100644 --- a/test/test_kernel_launch_checks.py +++ b/test/test_kernel_launch_checks.py @@ -1,4 +1,4 @@ -# Owner(s): ["module: unknown"] +# Owner(s): ["module: cuda"] from torch.testing._internal.common_utils import TestCase, run_tests from torch.testing._internal.check_kernel_launches import ( diff --git a/test/test_numba_integration.py b/test/test_numba_integration.py index f42dd4176daf..a194c2da6de3 100644 --- a/test/test_numba_integration.py +++ b/test/test_numba_integration.py @@ -1,4 +1,4 @@ -# Owner(s): ["module: unknown"] +# Owner(s): ["module: cuda"] import unittest