File tree Expand file tree Collapse file tree 2 files changed +5
-6
lines changed
pytorch_lightning/profiler Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,10 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
32
32
33
33
- Fixed ` SingleTPU ` calling ` all_gather ` ([ #6296 ] ( https://github.com/PyTorchLightning/pytorch-lightning/pull/6296 ) )
34
34
35
+
36
+ - Ensure we check deepspeed/sharded in multinode DDP ([ #6297 ] ( https://github.com/PyTorchLightning/pytorch-lightning/pull/6297 )
37
+
38
+
35
39
## [ 1.2.2] - 2021-03-02
36
40
37
41
### Added
Original file line number Diff line number Diff line change 13
13
# limitations under the License.
14
14
"""Profiler to check if there are any bottlenecks in your code."""
15
15
import cProfile
16
- import inspect
17
16
import io
18
17
import os
19
18
import pstats
20
19
import time
21
20
from abc import ABC , abstractmethod
22
21
from collections import defaultdict
23
22
from contextlib import contextmanager
24
- from typing import List , Optional , Union
23
+ from typing import Optional , Union
25
24
26
25
import numpy as np
27
- import torch
28
26
29
27
from pytorch_lightning import _logger as log
30
- from pytorch_lightning .utilities import rank_zero_only
31
28
from pytorch_lightning .utilities .cloud_io import get_filesystem
32
- from pytorch_lightning .utilities .distributed import rank_zero_warn
33
- from pytorch_lightning .utilities .exceptions import MisconfigurationException
34
29
35
30
36
31
class BaseProfiler (ABC ):
You can’t perform that action at this time.
0 commit comments