提交 bc92ddec authored 作者: Pascal Lamblin's avatar Pascal Lamblin

flake8

上级 05de6eca
...@@ -9,30 +9,29 @@ ProfileStats object for runtime and memory profiling. ...@@ -9,30 +9,29 @@ ProfileStats object for runtime and memory profiling.
# #
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import logging
__authors__ = "James Bergstra"
__reviewer__ = "Razvan Pascanu"
__copyright__ = "(c) 2011, Universite de Montreal"
__license__ = "3-clause BSD License"
__contact__ = "theano-dev <theano-dev@googlegroups.com>"
__docformat__ = "restructuredtext en"
import atexit import atexit
import copy import copy
import logging
import operator import operator
import os import os
import sys import sys
import time import time
from collections import defaultdict from collections import defaultdict
from six import iteritems
import numpy as np import numpy as np
import theano import theano
from six import iteritems
from theano.gof import graph from theano.gof import graph
__authors__ = "James Bergstra"
__reviewer__ = "Razvan Pascanu"
__copyright__ = "(c) 2011, Universite de Montreal"
__license__ = "3-clause BSD License"
__contact__ = "theano-dev <theano-dev@googlegroups.com>"
__docformat__ = "restructuredtext en"
logger = logging.getLogger('theano.compile.profiling') logger = logging.getLogger('theano.compile.profiling')
theano_imported_time = time.time() theano_imported_time = time.time()
...@@ -62,7 +61,7 @@ def _atexit_print_fn(): ...@@ -62,7 +61,7 @@ def _atexit_print_fn():
# Reverse sort in the order of compile+exec time # Reverse sort in the order of compile+exec time
for ps in sorted(_atexit_print_list, for ps in sorted(_atexit_print_list,
key=lambda a:a.compile_time + a.fct_call_time)[::-1]: key=lambda a: a.compile_time + a.fct_call_time)[::-1]:
if (ps.fct_callcount >= 1 or ps.compile_time > 1 or if (ps.fct_callcount >= 1 or ps.compile_time > 1 or
getattr(ps, 'callcount', 0) > 1): getattr(ps, 'callcount', 0) > 1):
ps.summary(file=destination_file, ps.summary(file=destination_file,
...@@ -116,6 +115,7 @@ def _atexit_print_fn(): ...@@ -116,6 +115,7 @@ def _atexit_print_fn():
if config.print_global_stats: if config.print_global_stats:
print_global_stats() print_global_stats()
def print_global_stats(): def print_global_stats():
""" """
Print the following stats: Print the following stats:
...@@ -133,7 +133,7 @@ def print_global_stats(): ...@@ -133,7 +133,7 @@ def print_global_stats():
else: else:
destination_file = open(config.profiling.destination, 'w') destination_file = open(config.profiling.destination, 'w')
print('='*50, file=destination_file) print('=' * 50, file=destination_file)
print('Global stats: ', print('Global stats: ',
'Time elasped since Theano import = %6.3fs, ' 'Time elasped since Theano import = %6.3fs, '
'Time spent in Theano functions = %6.3fs, ' 'Time spent in Theano functions = %6.3fs, '
...@@ -144,7 +144,7 @@ def print_global_stats(): ...@@ -144,7 +144,7 @@ def print_global_stats():
total_graph_opt_time, total_graph_opt_time,
total_time_linker), total_time_linker),
file=destination_file) file=destination_file)
print('='*50, file=destination_file) print('=' * 50, file=destination_file)
class ProfileStats(object): class ProfileStats(object):
...@@ -729,7 +729,7 @@ class ProfileStats(object): ...@@ -729,7 +729,7 @@ class ProfileStats(object):
file=file) file=file)
for node, t in sorted(self.linker_make_thunk_time.items(), for node, t in sorted(self.linker_make_thunk_time.items(),
key=operator.itemgetter(1))[::-1][:5]: key=operator.itemgetter(1))[::-1][:5]:
print(' Node %s time %es' % (node, t), print(' Node %s time %es' % (node, t),
file=file) file=file)
print('', file=file) print('', file=file)
...@@ -1132,7 +1132,7 @@ class ProfileStats(object): ...@@ -1132,7 +1132,7 @@ class ProfileStats(object):
# Store the max of some stats by any function in this profile. # Store the max of some stats by any function in this profile.
max_sum_size = max(max_sum_size, sum_size) max_sum_size = max(max_sum_size, sum_size)
def compute_max_stats(running_memory, stats): def compute_max_stats(running_memory, stats):
(max_node_memory_size, (max_node_memory_size,
max_running_max_memory_size, max_running_max_memory_size,
......
...@@ -37,7 +37,6 @@ whitelist_flake8 = [ ...@@ -37,7 +37,6 @@ whitelist_flake8 = [
"__init__.py", "__init__.py",
"tests/__init__.py", "tests/__init__.py",
"compile/__init__.py", "compile/__init__.py",
"compile/profiling.py",
"compile/sandbox/__init__.py", "compile/sandbox/__init__.py",
"compile/tests/__init__.py", "compile/tests/__init__.py",
"gpuarray/__init__.py", "gpuarray/__init__.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论