Golang InitGPU 예제들

예제 #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)
	}

}