func main() { nimble.Init() defer nimble.Cleanup() nimble.SetOD("test4.out") N0, N1, N2 := 1, 32, 128 cx, cy, cz := 3e-9, 3.125e-9, 3.125e-9 mesh := nimble.NewMesh(N0, N1, N2, cx, cy, cz) fmt.Println("mesh:", mesh) m := nimble.MakeChanN(3, "m", "", mesh, nimble.UnifiedMemory, 0) fmt.Println("m:", m) acc := 8 kernel := mag.BruteKernel(mesh, acc) B := gpu.NewConvolution("B", "T", mesh, nimble.UnifiedMemory, kernel, m).Output() const Bsat = 1.0053 const aex = mag.Mu0 * 13e-12 / Bsat exch := cpu.NewExchange6("Bex", "T", nimble.UnifiedMemory, m.NewReader(), aex) Bex := exch.Output() Beff := cpu.NewSum("Beff", B, Bex, Bsat, 1, nimble.UnifiedMemory).Output() const alpha = 1 tbox := cpu.NewLLGTorque("torque", m, Beff, alpha) nimble.Stack(tbox) torque := tbox.Output() const dt = 100e-15 solver := cpu.NewEuler(m, torque.NewReader(), mag.Gamma0, dt) M := cpu.Host(m.ChanN().UnsafeData()) for i := range M[2] { M[2][i] = 1 M[1][i] = 0.1 } every := 100 uni.Autosave(B, every, cpu.CPUDevice) uni.Autosave(m, every, cpu.CPUDevice) uni.Autosave(Bex, every, cpu.CPUDevice) uni.Autosave(Beff, every, cpu.CPUDevice) uni.Autosave(torque, every, cpu.CPUDevice) uni.Autotable(m, every, cpu.CPUDevice) nimble.RunStack() solver.Steps(100) res := cpu.Host(m.ChanN().UnsafeData()) got := [3]float32{res[0][0], res[1][0], res[2][0]} expect := [3]float32{-0.03450077, 0.21015842, 0.9770585} fmt.Println("result:", got) if got != expect { fmt.Println("expected:", expect) os.Exit(2) } else { fmt.Println("OK") } }
func main() { nimble.Init() defer nimble.Cleanup() nimble.SetOD("gpueuler.out") N0, N1, N2 := 1, 32, 128 cx, cy, cz := 3e-9, 3.125e-9, 3.125e-9 mesh := nimble.NewMesh(N0, N1, N2, cx, cy, cz) fmt.Println("mesh:", mesh) m := nimble.MakeChanN(3, "m", "", mesh, nimble.UnifiedMemory, 0) fmt.Println("m:", m) acc := 8 kernel := mag.BruteKernel(mesh, acc) B := conv.NewSymm2D("B", "T", mesh, nimble.UnifiedMemory, kernel, m).Output() const ( Bsat = 1.0053 aex = mag.Mu0 * 13e-12 / Bsat α = 1 ) //exch := cpu.NewExchange6("Bex", "T", nimble.UnifiedMemory, m.NewReader(), aex) exch := gpu.NewExchange6("Bex", m, aex) nimble.Stack(exch) Bex := exch.Output() // heff := MakeChan3("Heff", "", mesh) Beff := gpu.NewSum("Beff", B, Bex, Bsat, 1, nimble.UnifiedMemory).Output() tBox := gpu.NewLLGTorque("torque", m, Beff, α) nimble.Stack(tBox) torque := tBox.Output() const dt = 100e-15 solver := gpu.NewEuler(m, torque, mag.Gamma0, dt) M := cpu.Host(m.ChanN().UnsafeData()) for i := range M[2] { M[2][i] = 1 M[1][i] = 0.1 } every := 100 nimble.Autosave(B, every) nimble.Autosave(m, every) nimble.Autosave(Bex, every) nimble.Autosave(Beff, every) nimble.Autosave(torque, every) nimble.Autotable(m, every) nimble.RunStack() solver.Steps(100) res := cpu.Host(m.ChanN().UnsafeData()) got := [3]float32{res[0][0], res[1][0], res[2][0]} expect := [3]float32{-0.033120323, 0.20761484, 0.9776498} solver.Steps(10000) fmt.Println("result:", got) if got != expect { fmt.Println("expected:", expect) os.Exit(2) } else { fmt.Println("OK") } }
// Standard problem 4 on GPU func main() { nimble.Init() defer nimble.Cleanup() nimble.SetOD("gpu4.out") mem := nimble.GPUMemory const ( N0, N1, N2 = 1, 32 * 2, 128 cx, cy, cz = 3e-9, 3.125e-9 / 2, 3.125e-9 Bsat = 1.0053 Aex_red = mag.Mu0 * 13e-12 / Bsat α = 1 ) mesh := nimble.NewMesh(N0, N1, N2, cx, cy, cz) fmt.Println("mesh:", mesh) // TODO: MakeChanN -> NewQuant() m := nimble.MakeChanN(3, "m", "", mesh, mem, 0) M := gpu.Device3(m.ChanN().UnsafeData()) M[0].Memset(0) M[1].Memset(0.1) M[2].Memset(0.99) acc := 10 kernel := mag.BruteKernel(mesh, acc) B := gpu.NewConvolution("B", "T", mesh, mem, kernel, m).Output() exch := gpu.NewExchange6("Bex", m, Aex_red) nimble.Stack(exch) Bex := exch.Output() BeffBox := gpu.NewSum("Beff", B, Bex, Bsat, 1, mem) Beff := BeffBox.Output() tBox := gpu.NewLLGTorque("torque", m, Beff, α) nimble.Stack(tBox) torque := tBox.Output() solver := gpu.NewHeun(m, torque, 1e-15, mag.Gamma0) every := 100 uni.Autosave(m, every, gpu.GPUDevice) uni.Autotable(m, every/10, gpu.GPUDevice) solver.Advance(0.3e-9) // res := cpu.Host(m.ChanN().UnsafeData()) // got := [3]float32{res[0][0], res[1][0], res[2][0]} // expect := [3]float32{1.090642e-06, 0.6730072, 0.739636} // fmt.Println("result:", got) // if got != expect { // fmt.Println("expected:", expect) // os.Exit(2) // } else { // fmt.Println("OK") // } const ( Bx = -24.6E-3 By = 4.3E-3 Bz = 0 ) Bext := gpu.RunConst("Bext", "T", mesh, mem, []float64{Bz, By, Bx}) BeffBox.MAdd(Bext, 1) tBox.SetAlpha(0.02) solver.Advance(1e-9) }