pytorch/torch/testing/__init__.py
Philip Meier 07d5d7b5cc move kernel launch checks from torch.testing to torch.testing._internal.check_kernel_launches (#60862)
Summary:
The fact that these functions are only used in a single test might be a good enough reason to move them to that module.

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

Reviewed By: H-Huang

Differential Revision: D31141354

Pulled By: mruberry

fbshipit-source-id: 6ce1f721b88620c5f46222ad1b942bc689f0a3e0
2021-09-29 00:39:22 -07:00

5 lines
153 B
Python

from ._core import * # noqa: F403
from ._asserts import * # noqa: F403
from ._creation import * # noqa: F403
from ._deprecated import * # noqa: F403