add multithreading

This commit is contained in:
2023-05-20 21:22:56 +07:00
parent 3041cde44a
commit 548f2663fa
4 changed files with 12 additions and 11 deletions

View File

@@ -2,7 +2,7 @@ module forward
using Flux.Optimise: apply!
using Statistics, Flux, Random, LinearAlgebra
using Statistics, Flux, Random, LinearAlgebra, JSON3
using GeneralUtils
using ..types, ..snn_utils
@@ -77,8 +77,8 @@ function (kfn::kfn_1)(m::model, input_data::AbstractVector)
kfn.firedNeurons_t0 = [n.z_t for n in kfn.neuronsArray] #TODO check if it is used?
#CHANGE Threads.@threads for n in kfn.neuronsArray
for n in kfn.neuronsArray
Threads.@threads for n in kfn.neuronsArray
# for n in kfn.neuronsArray
n(kfn)
end
@@ -88,8 +88,8 @@ function (kfn::kfn_1)(m::model, input_data::AbstractVector)
kfn.firedNeurons |> unique! # use for random new neuron connection
end
# Threads.@threads for n in kfn.outputNeuronsArray
for n in kfn.outputNeuronsArray
Threads.@threads for n in kfn.outputNeuronsArray
# for n in kfn.outputNeuronsArray
n(kfn)
end