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

added debug stack trace to tensor error

Browse Source
This commit is contained in:
sugarme 2023-07-28 18:06:53 +10:00
parent 3cd8d8560f
commit bdf252d831
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ts/error.go
View File

@ -5,6 +5,7 @@ import "C"
import ( import (
"fmt" "fmt"
"runtime/debug"
"unsafe" "unsafe"
lib "github.com/sugarme/gotch/libtch" lib "github.com/sugarme/gotch/libtch"
@ -41,7 +42,8 @@ func TorchErr() error {
cptr := (*C.char)(lib.GetAndResetLastErr()) cptr := (*C.char)(lib.GetAndResetLastErr())
errStr := ptrToString(cptr) errStr := ptrToString(cptr)
if errStr != "" { if errStr != "" {
return fmt.Errorf("Libtorch API Error: %v\n", errStr) trace := string(debug.Stack())
return fmt.Errorf("Libtorch API Error: %v\n%v\n", errStr, trace)
} }
return nil return nil