mirror of https://github.com/llvm/torch-mlir
103 lines
2.1 KiB
Python
103 lines
2.1 KiB
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 @binary_lt_
|
|
@import_global
|
|
def binary_lt_():
|
|
# CHECK: %[[A:.*]] = constant 1 : i64
|
|
# CHECK: %[[B:.*]] = constant 2 : i64
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare %[[A]] "Lt" %[[B]] : i64, i64
|
|
return x < y
|
|
|
|
# CHECK-LABEL: func @binary_gt_
|
|
@import_global
|
|
def binary_gt_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "Gt" {{.*}} : i64, i64
|
|
return x > y
|
|
|
|
# CHECK-LABEL: func @binary_lte_
|
|
@import_global
|
|
def binary_lte_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "LtE" {{.*}} : i64, i64
|
|
return x <= y
|
|
|
|
# CHECK-LABEL: func @binary_gte_
|
|
@import_global
|
|
def binary_gte_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "GtE" {{.*}} : i64, i64
|
|
return x >= y
|
|
|
|
# CHECK-LABEL: func @binary_eq_
|
|
@import_global
|
|
def binary_eq_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "Eq" {{.*}} : i64, i64
|
|
return x == y
|
|
|
|
# CHECK-LABEL: func @binary_neq_
|
|
@import_global
|
|
def binary_neq_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "NotEq" {{.*}} : i64, i64
|
|
return x != y
|
|
|
|
# CHECK-LABEL: func @binary_is_
|
|
@import_global
|
|
def binary_is_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "Is" {{.*}} : i64, i64
|
|
return x is y
|
|
|
|
# CHECK-LABEL: func @binary_is_not_
|
|
@import_global
|
|
def binary_is_not_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "IsNot" {{.*}} : i64, i64
|
|
return x is not y
|
|
|
|
# CHECK-LABEL: func @binary_in_
|
|
@import_global
|
|
def binary_in_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "In" {{.*}} : i64, i64
|
|
return x in y
|
|
|
|
# CHECK-LABEL: func @binary_not_in_
|
|
@import_global
|
|
def binary_not_in_():
|
|
x = 1
|
|
y = 2
|
|
# CHECK: {{.*}} = basicpy.binary_compare {{.*}} "NotIn" {{.*}} : i64, i64
|
|
return x not in y
|
|
|
|
# @import_global
|
|
# def short_circuit():
|
|
# x = 1
|
|
# y = 2
|
|
# z = 3
|
|
# return x < y < z
|
|
|