panic on parallel runner

[lwch]

my code like this:

package main

import (
	"fmt"

	. "gorgonia.org/gorgonia"
	"gorgonia.org/tensor"
)

var w = tensor.New(tensor.WithShape(2, 2), tensor.WithBacking(GlorotEtAlN32(1, 2, 2)))

var xs = tensor.New(tensor.WithShape(2, 2), tensor.WithBacking([]float32{1, 2, 3, 4}))
var ys = tensor.New(tensor.WithShape(2, 2), tensor.WithBacking([]float32{5, 6, 7, 8}))

func assert(err error) {
	if err != nil {
		panic(err)
	}
}

func worker(ch chan struct{}) {
	g := NewGraph()
	x := NewMatrix(g, tensor.Float32, WithShape(2, 2), WithName("x"))
	y := NewMatrix(g, tensor.Float32, WithShape(2, 2), WithName("y"))
	wx := NodeFromAny(g, w, WithName("w"))
	pred := Must(Mul(x, wx))
	// mse
	diff := Must(Sub(y, pred))
	sqDiff := Must(Square(diff))
	loss := Must(Mean(sqDiff))
	_, err := Grad(loss, wx)
	assert(err)
	// vm
	vm := NewTapeMachine(g, BindDualValues(wx))
	defer vm.Close()
	for {
		<-ch
		Let(x, xs)
		Let(y, ys)
		assert(vm.RunAll()) // panic on this more than 10 times
		vm.Reset()
	}
}

func main() {
	ch := make(chan struct{})
	for i := 0; i < 2; i++ {
		go worker(ch)
	}
	for i := 0; i < 10000; i++ {
		ch <- struct{}{}
	}
	fmt.Println(w)
}

when i run in 1 goroutine, it works, but panic on parallel:

panic: PC 31. Failed to execute instruction Aᵀ × B      [CPU0 CPU4]     CPU20   true    false   true: Happened while attempting to execute Aᵀ × B       [CPU0 CPU4]CPU20    true    false   true. Node is 18. Register was: 20 : linAlgBinOp () × (2, 2) error: MatMul requires both operands to be matrices. Got t's shape: (), other's shape: (2, 2)

goroutine 6 [running]:
main.assert(...)
        /home/lwch/src/tnn/test/main.go:17
main.worker(0x0?)
        /home/lwch/src/tnn/test/main.go:40 +0x3d7
created by main.main
        /home/lwch/src/tnn/test/main.go:48 +0x34
exit status 2

how can i run in parallel and share the same weights

[chewxy]

oh okay.. this usecase is not one that I've seen before. Let me try