Add ConnGrowth weight output

Author: Vanessa Arndorfer

Simulator/Connections/Neuro/ConnGrowth_d.cpp

@@ -77,6 +77,12 @@ void ConnGrowth::updateEdgesWeights(int numVertices, AllVertices &vertices, AllE
 
    // copy device synapse count to host memory
    edges.copyDeviceEdgeCountsToHost(allEdgesDevice);
-   // copy device synapse summation coordinate to host memory
-   dynamic_cast<AllSpikingSynapses &>(edges).copyDeviceEdgeSumIdxToHost(allEdgesDevice);
+
+   // copy device synapse summation coordinate and weights to host memory
+   AllSpikingSynapses &synapses = dynamic_cast<AllSpikingSynapses &>(edges);
+   synapses.copyDeviceEdgeSumIdxToHost(allEdgesDevice);
+   synapses.copyDeviceEdgeWeightsToHost(allEdgesDevice);
+
+   // output weight matrix every epoch
+   synapses.outputWeights(simulator.getCurrentStep());
 }

Simulator/Core/GPUModel.cpp

@@ -237,7 +237,7 @@ void GPUModel::updateConnections()
    vertices.copyFromDevice(allVerticesDevice_);
 
    // Update Connections data
-   if (connections_->updateConnections(vertices)) {
+   if (connections_->updateConnections()) {
       connections_->updateEdgesWeights(Simulator::getInstance().getTotalVertices(), vertices, edges,
                                        allVerticesDevice_, allEdgesDevice_, getLayout());
       // create edge index map

Simulator/Edges/Neuro/AllSpikingSynapses.h

@@ -199,6 +199,12 @@ class AllSpikingSynapses : public AllNeuroEdges {
    ///  @param  allEdgesDevice  GPU address of the allEdges struct on device memory.
    void copyDeviceEdgeSumIdxToHost(void *allEdgesDevice);
 
+   ///  Get weights matrix in AllEdges struct on device memory.
+   ///
+   ///  @param  allEdgesDevice  GPU address of the AllSpikingSynapsesDeviceProperties struct
+   ///                             on device memory.
+   virtual void copyDeviceEdgeWeightsToHost(void *allEdgesDevice) ;
+
 protected:
    ///  Allocate GPU memories to store all synapses' states,
    ///  and copy them from host to GPU memory.

Simulator/Edges/Neuro/AllSpikingSynapses_d.cpp

@@ -273,6 +273,28 @@ void AllSpikingSynapses::copyDeviceEdgeCountsToHost(void *allEdgesDevice)
    //allEdges.countVertices_ = 0;
 }
 
+///  Get weights matrix in AllEdges struct on device memory.
+///
+///  @param  allEdgesDevice  GPU address of the AllSpikingSynapsesDeviceProperties struct
+///                             on device memory.
+void AllSpikingSynapses::copyDeviceEdgeWeightsToHost(void *allEdgesDevice)
+{
+   AllSpikingSynapsesDeviceProperties allEdgesDeviceProps;
+
+   int numVertices = Simulator::getInstance().getTotalVertices();
+   BGSIZE maxTotalSynapses = Simulator::getInstance().getMaxEdgesPerVertex() * numVertices;
+
+   HANDLE_ERROR(cudaMemcpy(&allEdgesDeviceProps, allEdgesDevice,
+                           sizeof(AllSpikingSynapsesDeviceProperties), cudaMemcpyDeviceToHost));
+
+   // std::cout << "size: " << vertexCount * vertexCount * sizeof(BGFLOAT) << std::endl;
+   // std::cout << "W_.data(): " << W_.data() << std::endl;
+   // std::cout << "allEdgesDeviceProps.W_: " << allEdgesDeviceProps.W_ << std::endl;
+
+   HANDLE_ERROR(cudaMemcpy(W_.data(), allEdgesDeviceProps.W_, 
+                  maxTotalSynapses * sizeof(BGFLOAT), cudaMemcpyDeviceToHost));
+}
+
 ///  Get summationCoord and in_use in AllEdges struct on device memory.
 ///
 ///  @param  allEdgesDevice  GPU address of the AllSpikingSynapsesDeviceProperties struct