add debug mode for create_xx_functions for detail time cost analysis

2023-05-08 15:42:25 +08:00
parent d4a75b9394
commit e201d03157
8 changed files with 70 additions and 38 deletions
--- a/algorithms/neat/genome/graph.py
+++ b/algorithms/neat/genome/graph.py
@@ -74,6 +74,7 @@ def topological_sort(nodes: Array, connections: Array) -> Array:
    return res


+@jit
@vmap
 def batch_topological_sort(pop_nodes: Array, pop_connections: Array) -> Array:
    """