update

Author: abhiutd

cbits.go

@@ -124,31 +124,31 @@ func (p *Predictor) Close() {
 }
 
 func (p *Predictor) StartProfiling(name, metadata string) error {
-  cname := C.CString(name)
-  cmetadata := C.CString(metadata)
-  defer C.free(unsafe.Pointer(cname))
-  defer C.free(unsafe.Pointer(cmetadata))
-  C.StartProfilingPytorch(p.ctx, cname, cmetadata)
-  return nil
+	cname := C.CString(name)
+	cmetadata := C.CString(metadata)
+	defer C.free(unsafe.Pointer(cname))
+	defer C.free(unsafe.Pointer(cmetadata))
+	C.StartProfilingPytorch(p.ctx, cname, cmetadata)
+	return nil
 
 }
 
 func (p *Predictor) EndProfiling() error {
-  C.EndProfilingPytorch(p.ctx)
-  return nil
+	C.EndProfilingPytorch(p.ctx)
+	return nil
 }
 
 func (p *Predictor) DisableProfiling() error {
-  C.DisableProfilingPytorch(p.ctx)
-  return nil
+	C.DisableProfilingPytorch(p.ctx)
+	return nil
 }
 
 func (p *Predictor) ReadProfile() (string, error) {
-  cstr := C.ReadProfilePytorch(p.ctx)
-  if cstr == nil {
-    return "", errors.New("failed to read nil profile")
-  }
-  defer C.free(unsafe.Pointer(cstr))
-  return C.GoString(cstr) nil
+	cstr := C.ReadProfilePytorch(p.ctx)
+	if cstr == nil {
+		return "", errors.New("failed to read nil profile")
+	}
+	defer C.free(unsafe.Pointer(cstr))
+	return C.GoString(cstr), nil
 
 }

init.go

@@ -1,18 +1,18 @@
 package pytorch
 
 import (
-  "github.com/rai-project/config"
-  "github.com/rai-project/logger"
-  "github.com/sirupsen/logrus"
+	"github.com/rai-project/config"
+	"github.com/rai-project/logger"
+	"github.com/sirupsen/logrus"
 )
 
 var (
-  log *logrus.Entry
+	log *logrus.Entry
 )
 
 func init() {
-  config.AfterInit(func() {
-    log = logger.New().WithField("pkg", "go-pytorch")
-  })
+	config.AfterInit(func() {
+		log = logger.New().WithField("pkg", "go-pytorch")
+	})
 
 }

version.go

@@ -1,7 +1,7 @@
 package pytorch
 
 var (
-  Version   = "0.0.1"
-  BuildDate = "undefined"
-  GitCommit = "undefined"
+	Version   = "0.0.1"
+	BuildDate = "undefined"
+	GitCommit = "undefined"
 )