mirror of https://github.com/llvm/torch-mlir
28 lines
1.2 KiB
MLIR
28 lines
1.2 KiB
MLIR
// RUN: npcomp-opt %s -canonicalize | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @torch.aten.__is__
|
|
// CHECK: %[[FALSE:.*]] = basicpy.bool_constant false
|
|
// CHECK: return %[[FALSE]] : !basicpy.BoolType
|
|
func @torch.aten.__is__(%arg0: !basicpy.ListType, %arg1: !basicpy.NoneType) -> !basicpy.BoolType{
|
|
%0 = torch.aten.__is__ %arg0, %arg1 : !basicpy.ListType, !basicpy.NoneType -> !basicpy.BoolType
|
|
return %0 : !basicpy.BoolType
|
|
}
|
|
// CHECK-LABEL: func @torch.aten.size(
|
|
// CHECK: %[[CM1:.*]] = constant -1 : i64
|
|
// CHECK: %[[C3:.*]] = constant 3 : i64
|
|
// CHECK: %[[RET:.*]] = basicpy.build_list %[[CM1]], %[[C3]] : (i64, i64) -> !basicpy.ListType
|
|
// CHECK: return %[[RET]] : !basicpy.ListType
|
|
func @torch.aten.size(%arg0: tensor<?x3xf32>) -> !basicpy.ListType {
|
|
%0 = torch.aten.size %arg0 : tensor<?x3xf32> -> !basicpy.ListType
|
|
return %0 : !basicpy.ListType
|
|
}
|
|
|
|
// CHECK-LABEL: func @torch.aten.len.t(
|
|
// CHECK: %[[LENGTH:.*]] = constant 2 : i64
|
|
// CHECK: return %[[LENGTH]] : i64
|
|
func @torch.aten.len.t(%arg0: i64) -> i64 {
|
|
%0 = basicpy.build_list %arg0, %arg0 : (i64, i64) -> !basicpy.ListType
|
|
%1 = torch.aten.len.t %0 : !basicpy.ListType -> i64
|
|
return %1 : i64
|
|
}
|