Exemplo n.º 1
0
 def test_check_output(self):
     if core.is_compiled_with_cuda():
         use_gpu = True
         self.check_output_with_option(use_gpu)
         self.assertTrue(
             PassVersionChecker.IsCompatible('tensorrt_subgraph_pass'))
Exemplo n.º 2
0
 def test_check_output(self):
     use_gpu = False
     self.check_output_with_option(use_gpu)
     self.assertTrue(PassVersionChecker.IsCompatible('mul_lstm_fuse_pass'))
 def test_pass_compatible(self):
     self.assertTrue(PassVersionChecker.IsCompatible(self.pass_name))
 def test_check_output(self):
     self.check_output()
     self.assertTrue(PassVersionChecker.IsCompatible(self.pass_name))