mirror of https://github.com/llvm/torch-mlir
33 lines
804 B
Python
33 lines
804 B
Python
|
# RUN: %PYTHON %s | npcomp-opt -split-input-file | FileCheck %s --dump-input=fail
|
||
|
|
|
||
|
|
from npcomp.compiler.frontend import *
|
||
|
|
|
||
|
|
|
||
|
|
def import_global(f):
|
||
|
|
fe = ImportFrontend()
|
||
|
|
fe.import_global_function(f)
|
||
|
|
print("// -----")
|
||
|
|
print(fe.ir_module.to_asm())
|
||
|
|
return f
|
||
|
|
|
||
|
|
# CHECK-LABEL: func @pass_no_return
|
||
|
|
@import_global
|
||
|
|
def pass_no_return():
|
||
|
|
# CHECK: %[[NONE:.*]] = basicpy.singleton : !basicpy.NoneType
|
||
|
|
# CHECK: %[[NONE_CAST:.*]] = basicpy.unknown_cast %[[NONE]] : !basicpy.NoneType -> !basicpy.UnknownType
|
||
|
|
# CHECK: return %[[NONE_CAST]]
|
||
|
|
# CHECK-NOT: return
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
# CHECK-LABEL: func @expr_statement
|
||
|
|
@import_global
|
||
|
|
def expr_statement():
|
||
|
|
# CHECK: basicpy.exec {
|
||
|
|
# CHECK: %[[V:.*]] = basicpy.binary_expr
|
||
|
|
# CHECK: basicpy.exec_discard %[[V]]
|
||
|
|
# CHECK: }
|
||
|
|
# CHECK: return
|
||
|
|
a = 1
|
||
|
|
a + 2
|