torch-mlir/test/Python/Compiler/constants.py

89 lines
2.1 KiB
Python
Raw Normal View History

# RUN: %PYTHON %s | npcomp-opt -split-input-file | FileCheck %s --dump-input=fail
from npcomp.compiler import test_config
import_global = test_config.create_import_dump_decorator()
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: 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
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: 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
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: func @bool_true_constant
2020-06-08 06:15:19 +08:00
@import_global
def bool_true_constant():
# CHECK: %[[A:.*]] = basicpy.bool_constant true
2020-06-08 06:15:19 +08:00
# CHECK: basicpy.unknown_cast %[[A]]
a = True
return a
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: func @bool_false_constant
2020-06-08 06:15:19 +08:00
@import_global
def bool_false_constant():
# CHECK: %[[A:.*]] = basicpy.bool_constant false
2020-06-08 06:15:19 +08:00
# CHECK: basicpy.unknown_cast %[[A]]
a = False
return a
2020-06-08 06:46:28 +08:00
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: func @string_constant
@import_global
def string_constant():
# CHECK: %[[A:.*]] = basicpy.str_constant "foobar"
# CHECK: basicpy.unknown_cast %[[A]]
a = "foobar"
return a
2020-06-08 06:46:28 +08:00
# CHECK-LABEL: func @joined_string_constant
@import_global
def joined_string_constant():
# CHECK: %[[A:.*]] = basicpy.str_constant "I am still here"
# CHECK: basicpy.unknown_cast %[[A]]
a = "I am" " still here"
return a
2020-06-08 07:00:29 +08:00
# CHECK-LABEL: func @bytes_constant
@import_global
def bytes_constant():
# CHECK: %[[A:.*]] = basicpy.bytes_constant "foobar"
# CHECK: basicpy.unknown_cast %[[A]]
a = b"foobar"
return a
2020-06-08 06:49:39 +08:00
# CHECK-LABEL: func @ellipsis
@import_global
def ellipsis():
# CHECK: %[[A:.*]] = basicpy.singleton : !basicpy.EllipsisType
# CHECK: basicpy.unknown_cast %[[A]]
a = ...
return a
2020-06-08 07:21:00 +08:00
2020-06-08 07:21:00 +08:00
# CHECK-LABEL: func @none_constant
@import_global
def none_constant():
# CHECK: %[[A:.*]] = basicpy.singleton : !basicpy.NoneType
# CHECK: basicpy.unknown_cast %[[A]]
a = None
return a