Copy variables to host after group_variable_set

brian2cuda/templates/group_variable_set.cu

@@ -20,6 +20,28 @@
     ///// endblock kernel_maincode /////
 {% endblock %}
 
+{% block extra_kernel_call_post %}
+    {# We need to copy modifed variables back to host in case they are used in
+       codeobjects that run on the host, which are synapse connect calls (e.g. in the
+       connect condition) and before run synapses push spikes, which initialized
+       synaptic variables.
+    #}
+    {% for var, varname in written_variables.items() %}
+    {% if var.dynamic %}
+    {{varname}} = dev{{varname}};
+    {% else %}
+    CUDA_SAFE_CALL(
+        cudaMemcpy(
+            {{varname}},
+            dev{{varname}},
+            sizeof({{c_data_type(var.dtype)}})*_num_{{varname}},
+            cudaMemcpyDeviceToHost
+        )
+    );
+    {% endif %}
+    {% endfor %}
+{% endblock %}
+
 {# _num_group_idx is defined in HOST_CONSTANTS, so we can't set _N before #}
 {% block define_N %}
 {% endblock %}

brian2cuda/templates/group_variable_set_conditional.cu

@@ -26,7 +26,11 @@
 
 
 {% block extra_kernel_call_post %}
-    {# We want to copy only those variables that were potentially modified in aboves kernel call. #}
+    {# We need to copy modifed variables back to host in case they are used in
+       codeobjects that run on the host, which are synapse connect calls (e.g. in the
+       connect condition) and before run synapses push spikes, which initialized
+       synaptic variables.
+    #}
     {% for var, varname in written_variables.items() %}
     {% if var.dynamic %}
     {{varname}} = dev{{varname}};