Python cond_brの例

コード例 #1

ファイル: test_py2and3.py プロジェクト: agramesh1/intel-quant-dialect

 def testCondBr(self):
     self.setUp()
     with self.module.function_context("foo", [self.boolType], []) as fun:
         with E.BlockContext() as blk1:
             E.ret([])
         with E.BlockContext([self.indexType]) as blk2:
             E.ret([])
         cst = E.constant_index(0)
         E.cond_br(fun.arg(0), blk1, [], blk2, [cst])
         printWithCurrentFunctionName(str(fun))

コード例 #2

ファイル: test_py2and3.py プロジェクト: zizhenG06/mlir

    def testCondBr(self):
        with self.module.function_context("foo", [self.boolType], []) as fun:
            with E.BlockContext() as blk1:
                E.ret([])
            with E.BlockContext([self.indexType]) as blk2:
                E.ret([])
            cst = E.constant_index(0)
            E.cond_br(fun.arg(0), blk1, [], blk2, [cst])

        code = str(fun)
        self.assertIn("cond_br %arg0, ^bb1, ^bb2(%c0 : index)", code)