torch-mlir/pytest/Compiler/constants.py

47 lines
1.1 KiB
Python
Raw Normal View History

# 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())
2020-06-08 06:15:19 +08:00
return f
# CHECK: func @integer_constants
@import_global
def integer_constants():
# CHECK: %[[A:.*]] = constant 100 : i64
a = 100
# CHECK: %[[A_CAST:.*]] = basicpy.unknown_cast %[[A]] : i64 -> !basicpy.UnknownType
# CHECK: return %[[A_CAST]]
return a
# CHECK: func @float_constants
@import_global
def float_constants():
# CHECK: %[[A:.*]] = constant 2.200000e+00 : f64
a = 2.2
# CHECK: %[[A_CAST:.*]] = basicpy.unknown_cast %[[A]] : f64 -> !basicpy.UnknownType
# CHECK: return %[[A_CAST]]
return a
2020-06-08 06:15:19 +08:00
# CHECK: func @bool_true_constant
@import_global
def bool_true_constant():
# CHECK: %[[A:.*]] = basicpy.bool_constant 1
# CHECK: basicpy.unknown_cast %[[A]]
a = True
return a
# CHECK: func @bool_false_constant
@import_global
def bool_false_constant():
# CHECK: %[[A:.*]] = basicpy.bool_constant 0
# CHECK: basicpy.unknown_cast %[[A]]
a = False
return a