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 (
"fmt"
"runtime/debug"
"unsafe"
lib "github.com/sugarme/gotch/libtch"
@ -41,7 +42,8 @@ func TorchErr() error {
cptr := (*C.char)(lib.GetAndResetLastErr())
errStr := ptrToString(cptr)
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