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/).
3232
3333- Fixed ` SingleTPU ` calling ` all_gather ` ([ #6296 ] ( https://github.com/PyTorchLightning/pytorch-lightning/pull/6296 ) )
3434
35+
36+ - Ensure we check deepspeed/sharded in multinode DDP ([ #6297 ] ( https://github.com/PyTorchLightning/pytorch-lightning/pull/6297 )
37+
38+
3539## [ 1.2.2] - 2021-03-02
3640
3741### Added
Original file line number Diff line number Diff line change 1313# limitations under the License.
1414"""Profiler to check if there are any bottlenecks in your code."""
1515import cProfile
16- import inspect
1716import io
1817import os
1918import pstats
2019import time
2120from abc import ABC , abstractmethod
2221from collections import defaultdict
2322from contextlib import contextmanager
24- from typing import List , Optional , Union
23+ from typing import Optional , Union
2524
2625import numpy as np
27- import torch
2826
2927from pytorch_lightning import _logger as log
30- from pytorch_lightning .utilities import rank_zero_only
3128from 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
3429
3530
3631class BaseProfiler (ABC ):
You can’t perform that action at this time.
0 commit comments