mirror of https://github.com/llvm/torch-mlir
96 lines
3.1 KiB
MLIR
96 lines
3.1 KiB
MLIR
// RUN: torch-mlir-opt %s -canonicalize | FileCheck %s
|
|
|
|
// CHECK-LABEL: func.func @torch_c.from_i1() -> !torch.bool {
|
|
// CHECK: %[[TRUE:.*]] = torch.constant.bool true
|
|
// CHECK: return %[[TRUE]] : !torch.bool
|
|
func.func @torch_c.from_i1() -> !torch.bool {
|
|
%c1_i1 = arith.constant true
|
|
%0 = torch_c.from_i1 %c1_i1
|
|
return %0 : !torch.bool
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.to_i1() -> i1 {
|
|
// CHECK: %[[C1_I1:.*]] = arith.constant true
|
|
// CHECK: return %[[C1_I1]] : i1
|
|
func.func @torch_c.to_i1() -> i1 {
|
|
%bool1 = torch.constant.bool true
|
|
%0 = torch_c.to_i1 %bool1
|
|
return %0 : i1
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.from_i64() -> !torch.int {
|
|
// CHECK: %[[INT5:.*]] = torch.constant.int 5
|
|
// CHECK: return %[[INT5]] : !torch.int
|
|
func.func @torch_c.from_i64() -> !torch.int {
|
|
%c5_i64 = arith.constant 5 : i64
|
|
%0 = torch_c.from_i64 %c5_i64
|
|
return %0 : !torch.int
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.to_i64() -> i64 {
|
|
// CHECK: %[[C5_I64:.*]] = arith.constant 5 : i64
|
|
// CHECK: return %[[C5_I64]] : i64
|
|
func.func @torch_c.to_i64() -> i64 {
|
|
%int5 = torch.constant.int 5
|
|
%0 = torch_c.to_i64 %int5
|
|
return %0 : i64
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.from_i64$to_i64() -> i64 {
|
|
// CHECK: %[[C5_I64:.*]] = arith.constant 5 : i64
|
|
// CHECK: return %[[C5_I64]] : i64
|
|
func.func @torch_c.from_i64$to_i64() -> i64 {
|
|
%c5_i64 = arith.constant 5 : i64
|
|
%0 = torch_c.from_i64 %c5_i64
|
|
%1 = torch_c.to_i64 %0
|
|
return %1 : i64
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.to_i64$from_i64() -> !torch.int {
|
|
// CHECK: %[[INT5:.*]] = torch.constant.int 5
|
|
// CHECK: return %[[INT5]] : !torch.int
|
|
func.func @torch_c.to_i64$from_i64() -> !torch.int {
|
|
%int5 = torch.constant.int 5
|
|
%0 = torch_c.to_i64 %int5
|
|
%1 = torch_c.from_i64 %0
|
|
return %1 : !torch.int
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.from_f64() -> !torch.float {
|
|
// CHECK: %[[FLOAT5:.*]] = torch.constant.float 5.000000e+00
|
|
// CHECK: return %[[FLOAT5]] : !torch.float
|
|
func.func @torch_c.from_f64() -> !torch.float {
|
|
%c5_f64 = arith.constant 5.000000e+00 : f64
|
|
%0 = torch_c.from_f64 %c5_f64
|
|
return %0 : !torch.float
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.to_f64() -> f64 {
|
|
// CHECK: %[[C5_f64:.*]] = arith.constant 5.000000e+00 : f64
|
|
// CHECK: return %[[C5_f64]] : f64
|
|
func.func @torch_c.to_f64() -> f64 {
|
|
%float5 = torch.constant.float 5.000000e+00
|
|
%0 = torch_c.to_f64 %float5
|
|
return %0 : f64
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.from_f64$to_f64() -> f64 {
|
|
// CHECK: %[[C5_f64:.*]] = arith.constant 5.000000e+00 : f64
|
|
// CHECK: return %[[C5_f64]] : f64
|
|
func.func @torch_c.from_f64$to_f64() -> f64 {
|
|
%c5_f64 = arith.constant 5.000000e+00 : f64
|
|
%0 = torch_c.from_f64 %c5_f64
|
|
%1 = torch_c.to_f64 %0
|
|
return %1 : f64
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @torch_c.to_f64$from_f64() -> !torch.float {
|
|
// CHECK: %[[FLOAT5:.*]] = torch.constant.float 5.000000e+00
|
|
// CHECK: return %[[FLOAT5]] : !torch.float
|
|
func.func @torch_c.to_f64$from_f64() -> !torch.float {
|
|
%float5 = torch.constant.float 5.000000e+00
|
|
%0 = torch_c.to_f64 %float5
|
|
%1 = torch_c.from_f64 %0
|
|
return %1 : !torch.float
|
|
}
|