autopep8

Author: lantiga

torch2c/__init__.py

@@ -7,21 +7,21 @@
 def _wrap(obj, prevfns=[]):
     if obj.__class__ not in emitters._class_map:
         raise Exception('%s does not have an Emitter' % obj.__class__)
-    return emitters._class_map[obj.__class__](obj,prevfns)
+    return emitters._class_map[obj.__class__](obj, prevfns)
 
 
 def _traverse_graph_recursive(out, el):
     if isinstance(el, Variable):
         var = _wrap(el)
     else:
         prevfns = []
-        if hasattr(el,'previous_functions'):
+        if hasattr(el, 'previous_functions'):
             prevfns = [f[0] for f in el.previous_functions]
-        var = _wrap(el,prevfns)
+        var = _wrap(el, prevfns)
     out.append(var)
     if hasattr(el, 'previous_functions'):
         for u in el.previous_functions:
-            _traverse_graph_recursive(out,u[0])
+            _traverse_graph_recursive(out, u[0])
 
 
 def _dedup_nodes(nodes):
@@ -37,10 +37,10 @@ def _dedup_nodes(nodes):
 
 def _traverse_graph(node):
     nodes = []
-    _traverse_graph_recursive(nodes,node.creator)
+    _traverse_graph_recursive(nodes, node.creator)
     nodes.reverse()
     nodes = _dedup_nodes(nodes)
-    var_dict = dict([(el.id,el) for el in nodes])
+    var_dict = dict([(el.id, el) for el in nodes])
     prev_none_count = 0
     while True:
         for el in nodes:
@@ -69,28 +69,29 @@ def _emit_c(nodes, out, fnname, out_path):
     # 3. free parameters (gen name from fnname)
     # to avoid loading from disk at each forward
     var_nodes = [el for el in nodes if type(el) == emitters.Variable]
-    out_node = _wrap_out_node(nodes,out)
+    out_node = _wrap_out_node(nodes, out)
     # TODO: make it more general re: last_node?
     last_node = nodes[-1]
-    ifndef = '#ifndef __%s__\n#define __%s__\n' % (2*(fnname.upper(),))
+    ifndef = '#ifndef __%s__\n#define __%s__\n' % (2 * (fnname.upper(),))
     endif = '#endif'
     includes = '#include "TH.h"\n#include "THNN.h"\n#include "torch2c.h"\n'
-    fndecl = 'void %s(%s)' % (fnname, 
-              ', '.join([el.emit_decl() for el in var_nodes + [out_node]]))
-    calls = [el.emit_call(out_path,'data') for el in nodes]
+    fndecl = 'void %s(%s)' % (fnname,
+                              ', '.join([el.emit_decl() for el in var_nodes + [out_node]]))
+    calls = [el.emit_call(out_path, 'data') for el in nodes]
     copy_out = out_node.emit_copy(last_node.id_var_name())
     # TODO: be smarter re: frees
     # analyze calls backwards and free right after last use
     frees = [el.emit_free() for el in nodes]
     frees.reverse()
     indent = ' ' * 2
-    lines = [indent + el for el in '\n'.join(calls + [copy_out] + frees).split('\n') if el]
+    lines = [
+        indent + el for el in '\n'.join(calls + [copy_out] + frees).split('\n') if el]
     lines = [ifndef, includes, fndecl, '{'] + lines + ['}', endif]
     return '\n'.join(lines)
 
 
 def _to_persisted(var_node):
-    persisted = emitters.PersistedVariable(var_node.obj,[])
+    persisted = emitters.PersistedVariable(var_node.obj, [])
     persisted.numtype = var_node.numtype
     return persisted
 
@@ -104,41 +105,45 @@ def _clone_var(var):
 
 
 def _emit_test(nodes, out, fnname, filename, out_path):
-    var_nodes = [_to_persisted(el) for el in nodes if type(el) == emitters.Variable]
-    out_node = _to_persisted(_wrap_out_node(nodes,out))
-    out_baseline_node = _to_persisted(_wrap_out_node(nodes,_clone_var(out)))
+    var_nodes = [_to_persisted(el)
+                 for el in nodes if type(el) == emitters.Variable]
+    out_node = _to_persisted(_wrap_out_node(nodes, out))
+    out_baseline_node = _to_persisted(_wrap_out_node(nodes, _clone_var(out)))
     out_node.obj.data.zero_()
     includes = '#include "%s"' % filename
     fndecl = 'int main(int argc, char *argv[])'
-    calls = [el.emit_call(out_path,'data') for el in var_nodes + [out_baseline_node, out_node]]
+    calls = [el.emit_call(out_path, 'data')
+             for el in var_nodes + [out_baseline_node, out_node]]
     fncall = '%s(%s);' % (fnname,
-                    ', '.join([el.id_var_name() for el in var_nodes + [out_node]]))
-    equal_var = '%s_equal_%s' % (out_node.id_var_name(), out_baseline_node.id_var_name())
-    equal = out_node.emit_equal(equal_var,out_baseline_node.id_var_name())
+                          ', '.join([el.id_var_name() for el in var_nodes + [out_node]]))
+    equal_var = '%s_equal_%s' % (
+        out_node.id_var_name(), out_baseline_node.id_var_name())
+    equal = out_node.emit_equal(equal_var, out_baseline_node.id_var_name())
     print_equal = 'printf("Test passed: %d\\n",' + equal_var + ');'
-    frees = [el.emit_free() for el in var_nodes + [out_baseline_node, out_node]]
+    frees = [el.emit_free()
+             for el in var_nodes + [out_baseline_node, out_node]]
     ret = 'return %s ? EXIT_SUCCESS : EXIT_FAILURE;' % equal_var
     indent = ' ' * 2
-    lines = [indent + el for el in '\n'.join(calls + [fncall, equal, print_equal] + frees + [ret]).split('\n') if el]
+    lines = [indent + el for el in '\n'.join(
+        calls + [fncall, equal, print_equal] + frees + [ret]).split('\n') if el]
     lines = [includes, fndecl, '{'] + lines + ['}']
     return '\n'.join(lines)
 
 
 def compile(node, fnname, out_path, compile_test=False):
-    includedir = os.path.join(os.path.dirname(__file__),'..','include')
+    includedir = os.path.join(os.path.dirname(__file__), '..', 'include')
     nodes = _traverse_graph(node)
     if not os.path.isdir(out_path):
         os.mkdir(out_path)
-    data_path = os.path.join(out_path,'data')
+    data_path = os.path.join(out_path, 'data')
     if not os.path.isdir(data_path):
         os.mkdir(data_path)
     filename = "%s.h" % fnname
-    src = _emit_c(nodes,node,fnname,out_path)
-    with open(os.path.join(out_path,filename),'w') as f:
+    src = _emit_c(nodes, node, fnname, out_path)
+    with open(os.path.join(out_path, filename), 'w') as f:
         f.write(src)
     if compile_test:
         test_filename = "%s_test.c" % fnname
-        test_src = _emit_test(nodes,node,fnname,filename,out_path)
-        with open(os.path.join(out_path,test_filename),'w') as f:
+        test_src = _emit_test(nodes, node, fnname, filename, out_path)
+        with open(os.path.join(out_path, test_filename), 'w') as f:
             f.write(test_src)
-