Commit df3e79f0 authored by Paul Schultz's avatar Paul Schultz
Browse files

cluster problem

parent c3e7c860
......@@ -164,7 +164,8 @@ from numba import njit, float64, complex128, void, int32
"""
def_network_rhs_string = """def network_rhs_numba(y, t):
def_network_rhs_string = """
def network_rhs_numba(y, t):
dydt = np.empty({total_length} + {length})
v = y[:{total_length}].view(np.complex128)
......@@ -185,9 +186,9 @@ from numba import njit, float64, complex128, void, int32
)
if cache_flag:
def_network_rhs_string = "@njit(float64[:](float64[:], float64), cache=True)".join(def_network_rhs_string)
def_network_rhs_string = "@njit(float64[:](float64[:], float64), cache=True)" + def_network_rhs_string
else:
def_network_rhs_string = "@njit(float64[:](float64[:], float64))".join(def_network_rhs_string)
def_network_rhs_string = "@njit(float64[:](float64[:], float64))" + def_network_rhs_string
def_network_rhs_string += "".join([node.node_dynamics_string(j) for j, node in enumerate(node_list)])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment