2021-06-21 16:51:53 +00:00
|
|
|
from pytorch_lightning.utilities import rank_zero_deprecation
|
2021-06-04 15:38:29 +00:00
|
|
|
|
|
|
|
rank_zero_deprecation(
|
2021-06-21 16:51:53 +00:00
|
|
|
"Using ``import pytorch_lightning.profiler.profilers`` is deprecated in v1.4, and will be removed in v1.6. "
|
2021-06-04 15:38:29 +00:00
|
|
|
"HINT: Use ``import pytorch_lightning.profiler`` directly."
|
|
|
|
)
|
|
|
|
|
2021-08-02 16:05:56 +00:00
|
|
|
from pytorch_lightning.profiler.advanced import AdvancedProfiler # noqa: E402
|
|
|
|
from pytorch_lightning.profiler.base import AbstractProfiler, BaseProfiler, PassThroughProfiler # noqa: E402
|
|
|
|
from pytorch_lightning.profiler.pytorch import PyTorchProfiler # noqa: E402
|
|
|
|
from pytorch_lightning.profiler.simple import SimpleProfiler # noqa: E402
|
|
|
|
from pytorch_lightning.profiler.xla import XLAProfiler # noqa: E402
|
2021-06-04 15:38:29 +00:00
|
|
|
|
|
|
|
__all__ = [
|
2021-07-26 11:37:35 +00:00
|
|
|
"AbstractProfiler",
|
|
|
|
"BaseProfiler",
|
|
|
|
"AdvancedProfiler",
|
|
|
|
"PassThroughProfiler",
|
|
|
|
"PyTorchProfiler",
|
|
|
|
"SimpleProfiler",
|
|
|
|
"XLAProfiler",
|
2021-06-04 15:38:29 +00:00
|
|
|
]
|