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'))
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))