// cu.Init(), but error is fatal and does not dump stack. func tryCuInit() { defer func() { err := recover() util.FatalErr(err, "initialize GPU:") }() cu.Init(0) }
func InitCuda() { runtime.LockOSThread() cu.Init(0) cu.CtxCreate(cu.CTX_SCHED_AUTO, 0).SetCurrent() }
// needed for all other tests. func init() { cu.Init(0) ctx := cu.CtxCreate(cu.CTX_SCHED_AUTO, 0) cu.CtxSetCurrent(ctx) fmt.Println("Created CUDA context") }