// 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")
}
