From bdf252d8316c387cd9f2209d2348566c94264918 Mon Sep 17 00:00:00 2001 From: sugarme Date: Fri, 28 Jul 2023 18:06:53 +1000 Subject: [PATCH] added debug stack trace to tensor error --- ts/error.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ts/error.go b/ts/error.go index 9bcdc5c..821383c 100644 --- a/ts/error.go +++ b/ts/error.go @@ -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