Skip to content

Commit fb5bb3f

Browse files
fixed cudaMemcpy bugs
1 parent 8f1b4a8 commit fb5bb3f

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

Simulator/Core/GPUModel.cpp

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -178,16 +178,14 @@ void GPUModel::advance()
178178
std::vector<float> sp_h(verts);
179179
std::vector<float> vm_h(verts);
180180
std::vector<float> Inoise_h(verts);
181-
HANDLE_ERROR(cudaMemcpy(sp_h.data(), allVerticesDevice_->summationPoints_, verts * sizeof(float),
181+
AllIFNeuronsDeviceProperties validationNeurons;
182+
HANDLE_ERROR(cudaMemcpy((void *)&validationNeurons, allVerticesDevice_,
183+
sizeof(AllIFNeuronsDeviceProperties), cudaMemcpyDeviceToHost));
184+
HANDLE_ERROR(cudaMemcpy(sp_h.data(), validationNeurons.spValidation_, verts * sizeof(float),
182185
cudaMemcpyDeviceToHost));
183-
HANDLE_ERROR(
184-
cudaMemcpy(sp_h.data(), randNoise_d, verts * sizeof(float), cudaMemcpyDeviceToHost));
185-
HANDLE_ERROR(cudaMemcpy(vm_h.data(), ((AllIFNeuronsDeviceProperties *)(allVerticesDevice_))->Vm_,
186-
verts * sizeof(float), cudaMemcpyDeviceToHost));
187-
HANDLE_ERROR(cudaMemcpy(Inoise_h.data(),
188-
((AllIFNeuronsDeviceProperties *)(allVerticesDevice_))->Inoise_,
189-
verts * sizeof(float), cudaMemcpyDeviceToHost));
190-
HANDLE_ERROR(cudaMemcpy(sp_h.data(), allVerticesDevice_->spValidation_, verts * sizeof(float),
186+
HANDLE_ERROR(cudaMemcpy(vm_h.data(), validationNeurons.Vm_, verts * sizeof(float),
187+
cudaMemcpyDeviceToHost));
188+
HANDLE_ERROR(cudaMemcpy(Inoise_h.data(), validationNeurons.Inoise_, verts * sizeof(float),
191189
cudaMemcpyDeviceToHost));
192190

193191
for (int i = 0; i < verts; i++) {

0 commit comments

Comments
 (0)