// 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
}