message: fix: Fix for torch scripted module faiure with DLFW
diff: @@ -30,6 +30,7 @@ class TestCompile(ModelTestCase):
  self.assertTrue(same < 2e-2)
  def test_compile_script(self):
  + with torch.no_grad():
  trt_mod = torchtrt.ts.compile(self.scripted_model,
  inputs=[self.input],
  device=torchtrt.Device(gpu_id=0),
  @@ -46,6 +47,7 @@ class TestCompile(ModelTestCase):
  self.assertTrue(same < 2e-2)
  def test_compile_global_nn_mod(self):
  + with torch.no_grad():
  trt_mod = torchtrt.compile(self.model,
  inputs=[self.input],
  device=torchtrt.Device(gpu_id=0),
  