26
26
from .profilers .cpu_profiler import CPUProfiler
27
27
from .profilers .allocation_profiler import AllocationProfiler
28
28
from .profilers .block_profiler import BlockProfiler
29
- from .profilers .tf_profiler import TFProfiler
30
29
31
30
32
31
class Span (object ):
33
32
34
33
def __init__ (self , stop_func = None ):
35
34
if stop_func :
36
35
self .stop_func = stop_func
36
+ else :
37
+ self .stop_func = None
37
38
38
39
39
40
def stop (self ):
@@ -51,7 +52,7 @@ def __exit__(self, exc_type, exc_value, traceback):
51
52
52
53
class Agent (object ):
53
54
54
- AGENT_VERSION = "1.2.4 "
55
+ AGENT_VERSION = "1.2.6 "
55
56
SAAS_DASHBOARD_ADDRESS = "https://agent-api.stackimpact.com"
56
57
57
58
def __init__ (self , ** kwargs ):
@@ -100,15 +101,6 @@ def __init__(self, **kwargs):
100
101
config .report_interval = 120
101
102
self .block_reporter = ProfileReporter (self , BlockProfiler (self ), config )
102
103
103
- config = ProfilerConfig ()
104
- config .log_prefix = 'TensorFlow profiler'
105
- config .max_profile_duration = 20
106
- config .max_span_duration = 5
107
- config .max_span_count = 30
108
- config .span_interval = 20
109
- config .report_interval = 120
110
- self .tf_reporter = ProfileReporter (self , TFProfiler (self ), config )
111
-
112
104
self .options = None
113
105
114
106
@@ -160,7 +152,6 @@ def start(self, **kwargs):
160
152
self .cpu_reporter .setup ()
161
153
self .allocation_reporter .setup ()
162
154
self .block_reporter .setup ()
163
- self .tf_reporter .setup ()
164
155
self .span_reporter .setup ()
165
156
self .error_reporter .setup ()
166
157
self .process_reporter .setup ()
@@ -211,7 +202,6 @@ def enable(self):
211
202
self .cpu_reporter .start ()
212
203
self .allocation_reporter .start ()
213
204
self .block_reporter .start ()
214
- self .tf_reporter .start ()
215
205
self .span_reporter .start ()
216
206
self .error_reporter .start ()
217
207
self .process_reporter .start ()
@@ -223,7 +213,6 @@ def disable(self):
223
213
self .cpu_reporter .stop ()
224
214
self .allocation_reporter .stop ()
225
215
self .block_reporter .stop ()
226
- self .tf_reporter .stop ()
227
216
self .span_reporter .stop ()
228
217
self .error_reporter .stop ()
229
218
self .process_reporter .stop ()
@@ -244,8 +233,6 @@ def profile(self, name='Default'):
244
233
active_reporters .append (self .allocation_reporter )
245
234
if self .block_reporter .started :
246
235
active_reporters .append (self .block_reporter )
247
- if self .tf_reporter .started :
248
- active_reporters .append (self .tf_reporter )
249
236
250
237
if len (active_reporters ) > 0 :
251
238
selected_reporter = active_reporters [int (math .floor (random .random () * len (active_reporters )))]
@@ -318,14 +305,6 @@ def stop_block_profiler(self):
318
305
self ._stop_profiler (self .block_reporter )
319
306
320
307
321
- def start_tf_profiler (self ):
322
- self ._start_profiler (self .tf_reporter )
323
-
324
-
325
- def stop_tf_profiler (self ):
326
- self ._stop_profiler (self .tf_reporter )
327
-
328
-
329
308
def destroy (self ):
330
309
if not self .agent_started :
331
310
self .log ('Agent has not been started' )
@@ -340,15 +319,13 @@ def destroy(self):
340
319
self .cpu_reporter .stop ()
341
320
self .allocation_reporter .stop ()
342
321
self .block_reporter .stop ()
343
- self .tf_reporter .stop ()
344
322
self .error_reporter .stop ()
345
323
self .span_reporter .stop ()
346
324
self .process_reporter .stop ()
347
325
348
326
self .cpu_reporter .destroy ()
349
327
self .allocation_reporter .destroy ()
350
328
self .block_reporter .destroy ()
351
- self .tf_reporter .destroy ()
352
329
self .error_reporter .destroy ()
353
330
self .span_reporter .destroy ()
354
331
self .process_reporter .destroy ()
0 commit comments