From 7533f43c2e485759391e35664f5c8717a885f7fb Mon Sep 17 00:00:00 2001
From: pjongy <hi.pjongy@gmail.com>
Date: Thu, 19 Oct 2023 03:17:21 +0000
Subject: [PATCH] Fix concurrency issue in newTensor

---
 ts/tensor.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ts/tensor.go b/ts/tensor.go
index 33f6666..4e5531d 100644
--- a/ts/tensor.go
+++ b/ts/tensor.go
@@ -60,7 +60,6 @@ func newTensor(ctensor lib.Ctensor, nameOpt ...string) *Tensor {
 	if len(nameOpt) == 0 {
 		nameOpt = []string{}
 	}
-	name := newName(nameOpt...)
 
 	x := new(Tensor)
 	x.ctensor = ctensor
@@ -70,6 +69,7 @@ func newTensor(ctensor lib.Ctensor, nameOpt ...string) *Tensor {
 	nbytes := x.nbytes()
 	atomic.AddInt64(&AllocatedMem, nbytes)
 	lock.Lock()
+	name := newName(nameOpt...)
 	if _, ok := ExistingTensors[name]; ok {
 		name = fmt.Sprintf("%s_%09d", name, TensorCount)
 	}