mirror of
https://github.com/unknwon/the-way-to-go_ZH_CN.git
synced 2025-08-11 23:52:31 +08:00
b242141e22
* 更新14、15和17章目录 * Update README.md * 添加章节序号 * 添加章节序号 * 添加代码链接 * 修改错别字 * 删除多余字符 * 修复代码与文件不对应的问题 * 修改代码
43 lines
1.6 KiB
Markdown
43 lines
1.6 KiB
Markdown
# 14.13 在多核心上并行计算
|
||
|
||
假设我们有`NCPU`个CPU核心：`const NCPU = 4 //对应一个四核处理器` 然后我们想把计算量分成`NCPU`个部分，每一个部分都和其他部分并行运行。
|
||
|
||
这可以通过以下代码所示的方式完成（我们且省略具体参数）
|
||
|
||
```go
|
||
func DoAll(){
|
||
sem := make(chan int, NCPU) // Buffering optional but sensible
|
||
for i := 0; i < NCPU; i++ {
|
||
go DoPart(sem)
|
||
}
|
||
// Drain the channel sem, waiting for NCPU tasks to complete
|
||
for i := 0; i < NCPU; i++ {
|
||
<-sem // wait for one task to complete
|
||
}
|
||
// All done.
|
||
}
|
||
|
||
func DoPart(sem chan int) {
|
||
// do the part of the computation
|
||
sem <-1 // signal that this piece is done
|
||
}
|
||
|
||
func main() {
|
||
runtime.GOMAXPROCS(NCPU) // runtime.GOMAXPROCS = NCPU
|
||
DoAll()
|
||
}
|
||
|
||
```
|
||
- `DoAll()`函数创建了一个`sem`通道，每个并行计算都将在对其发送完成信号；在一个 for 循环中`NCPU`个协程被启动了，每个协程会承担`1/NCPU`的工作量。每一个`DoPart()`协程都会向`sem`通道发送完成信号。
|
||
|
||
- `DoAll()`会在 for 循环中等待`NCPU`个协程完成：`sem`通道就像一个信号量，这份代码展示了一个经典的信号量模式。（参见 [14.2.7](14.2.md#1427-%E4%BF%A1%E5%8F%B7%E9%87%8F%E6%A8%A1%E5%BC%8F)）
|
||
|
||
在以上运行模型中，您还需将`GOMAXPROCS`设置为`NCPU`（参见 [14.1.3](14.1.md#1413-%E4%BD%BF%E7%94%A8-gomaxprocs)）。
|
||
|
||
|
||
## 链接
|
||
|
||
- [目录](directory.md)
|
||
- 上一节：[协程链](14.12.md)
|
||
- 下一节：[并行化大量数据的计算](14.14.md)
|