Golang InitGPU示例

说明

golang initgpu示例是从最受好评的开源项目中提取的实现代码，你可以参考下面示例的使用方式。

编程语言: Golang

命名空间/包名称: hotspin-core/gpu

示例#1

文件： main.go 项目： LStoleriu/hotspin

// based on the -gpu flag, activate set of GPUs to use
func initMultiGPU() {
	flag := *flag_gpus
	cuFlags := parseCuFlags()

	gpus := Atoi(flag)
	gpu.InitGPU(gpus, cuFlags)
}

示例#2

文件： texgen.go 项目： LStoleriu/hotspin

// INTERNAL: generates LaTeX files with documentation.
func TexGen() {
	initCUDA()
	gpu.InitGPU(0, 0)

	out := OpenWRONLY("modules.tex")
	defer out.Close()

	// order by alphabetic order
	keys := make([]string, len(modules))
	i := 0
	for k, _ := range modules {
		keys[i] = k
		i++
	}
	sort.Strings(keys)

	for _, name := range keys {
		moduleTexGen(out, name)
	}

}