andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix build error

Browse Source
This commit is contained in:
pjongy 2023-10-19 04:18:26 +00:00
parent 8031bb85f5
commit 6f7d68572b
2 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ts/scalar.go
View File

@ -20,6 +20,8 @@ func freeCScalar(x *Scalar) error {
if gotch.Debug {
nbytes := x.nbytes()
atomic.AddInt64(&AllocatedMem, -nbytes)
log.Printf("INFO: Released scalar %q - C memory: %d bytes.\n", x.name, nbytes)
}
lock.Lock()
delete(ExistingScalars, x.name)
@ -30,10 +32,6 @@ func freeCScalar(x *Scalar) error {
return err
}
if gotch.Debug {
log.Printf("INFO: Released scalar %q - C memory: %d bytes.\n", x.name, nbytes)
}
return nil
}
@ -58,15 +56,13 @@ func newScalar(cscalar lib.Cscalar, nameOpt ...string) *Scalar {
if gotch.Debug {
nbytes := x.nbytes()
atomic.AddInt64(&AllocatedMem, nbytes)
log.Printf("INFO: scalar %q added - Allocated memory (%d bytes).\n", x.name, nbytes)
}
lock.Lock()
ExistingScalars[x.name] = struct{}{}
lock.Unlock()
if gotch.Debug {
log.Printf("INFO: scalar %q added - Allocated memory (%d bytes).\n", x.name, nbytes)
}
runtime.SetFinalizer(x, freeCScalar)
return x

6
ts/tensor.go
View File

@ -70,6 +70,8 @@ func newTensor(ctensor lib.Ctensor, nameOpt ...string) *Tensor {
if gotch.Debug {
nbytes := x.nbytes()
atomic.AddInt64(&AllocatedMem, nbytes)
log.Printf("INFO: Added tensor %q - Allocated memory: %d bytes.\n", x.name, nbytes)
}
lock.Lock()
if _, ok := ExistingTensors[name]; ok {
@ -80,10 +82,6 @@ func newTensor(ctensor lib.Ctensor, nameOpt ...string) *Tensor {
x.name = name
if gotch.Debug {
log.Printf("INFO: Added tensor %q - Allocated memory: %d bytes.\n", x.name, nbytes)
}
x.calledFrom = "newTensor()"
runtime.SetFinalizer(x, freeCTensor)