PSO on GPU
- 0 Collaborators
利用openmp和mpi实现粒子群优化算法在GPU上的平行化运算,能高效处理大量粒子的运算。 ...learn more
Project status: Under Development
Overview / Usage
粒子群优化算法利用粒子群在搜索空间内移动寻找最优解,在每一轮迭代中系统选出最优粒子(优化函数值最接近0),是其余粒子向其靠拢,直至达到迭代次数或是优化函数值达到0。,
Methodology / Approach
在每一轮迭代里,粒子的移动向量只与历届最优粒子以及自身位置有关,因此粒子的位置与移动向量都适合进行平行运算,每一个线程负责一部分粒子的移动向量与位置更新。一轮迭代结束之后线程会进行进行通信找出最优粒子,将其位置进行复制之后开始下一轮训练。
Technologies Used
MPI, openmp
Repository
https://github.com/herokungugly/PSO_c/blob/main/PSO_gpu.ipynb