Add NoGradGuard to libtorch execution (#263)

lantiga · web-flow · commit 9e09de759783 · 2020-01-20T22:23:34.000+01:00
diff --git a/src/libtorch_c/torch_c.cpp b/src/libtorch_c/torch_c.cpp
@@ -218,10 +218,12 @@ void torchRunModule(ModuleContext* ctx, const char* fnName,
   }
 
   if (ctx->module) {
+    torch::NoGradGuard guard;
     torch::jit::script::Method method = ctx->module->get_method(fnName);
     method.run(stack);
   }
   else {
+    torch::NoGradGuard guard;
     torch::jit::Function& fn = ctx->cu->get_function(fnName);
     fn.run(stack);
   }

Original file line number	Diff line number	Diff line change
`@@ -218,10 +218,12 @@ void torchRunModule(ModuleContext* ctx, const char* fnName,`
`218`	`218`	`}`
`219`	`219`
`220`	`220`	`if (ctx->module) {`
	`221`	`+ torch::NoGradGuard guard;`
`221`	`222`	`torch::jit::script::Method method = ctx->module->get_method(fnName);`
`222`	`223`	`method.run(stack);`
`223`	`224`	`}`
`224`	`225`	`else {`
	`226`	`+ torch::NoGradGuard guard;`
`225`	`227`	`torch::jit::Function& fn = ctx->cu->get_function(fnName);`
`226`	`228`	`fn.run(stack);`
`227`	`229`	`}`