Merge pull request #125 from sony/hotfix/20180403_code_generator_bug

code generator random order bug
sony · Apr 17, 2018 · d7d7918 · d7d7918
2 parents 96f141e + 5d6ae34
commit d7d7918
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/nbla/function/function_impl.hpp.tmpl b/include/nbla/function/function_impl.hpp.tmpl
@@ -14,12 +14,13 @@
 
 <%
 from utils.type_conv import type_from_proto
+from collections import OrderedDict
 dec_targs = ', '.join(['typename ' + t for t in ttypes])
 targs = ', '.join(ttypes)
 args = ''.join([', ' + k for k, v in arguments.items()])
 args_ = ''.join([', ' + k + '_' for k, v in arguments.items()])
 
-arg_types = {k: type_from_proto[v['type']]['cpp'] for k, v in arguments.items()}
+arg_types = OrderedDict([(k, type_from_proto[v['type']]['cpp']) for k, v in arguments.items()])
 dec_args = ''.join([', %s %s' % (v, k) for k, v in arg_types.items()])
 
 def get_min(inputs):