PSO on GPU

stephen yeh

stephen yeh

Taipei City

0 0
  • 0 Collaborators

利用openmp和mpi实现粒子群优化算法在GPU上的平行化运算,能高效处理大量粒子的运算。 ...learn more

Project status: Under Development

Artificial Intelligence

Intel Technologies
DevCloud, oneAPI

Code Samples [1]

Overview / Usage

粒子群优化算法利用粒子群在搜索空间内移动寻找最优解,在每一轮迭代中系统选出最优粒子(优化函数值最接近0),是其余粒子向其靠拢,直至达到迭代次数或是优化函数值达到0。,

Methodology / Approach

在每一轮迭代里,粒子的移动向量只与历届最优粒子以及自身位置有关,因此粒子的位置与移动向量都适合进行平行运算,每一个线程负责一部分粒子的移动向量与位置更新。一轮迭代结束之后线程会进行进行通信找出最优粒子,将其位置进行复制之后开始下一轮训练。

Technologies Used

MPI, openmp

Repository

https://github.com/herokungugly/PSO_c/blob/main/PSO_gpu.ipynb

Comments (0)