File: addition.mlir

package info (click to toggle)
swiftlang 6.1.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,791,604 kB
  • sloc: cpp: 9,901,740; ansic: 2,201,431; asm: 1,091,827; python: 308,252; objc: 82,166; f90: 80,126; lisp: 38,358; pascal: 25,559; sh: 20,429; ml: 5,058; perl: 4,745; makefile: 4,484; awk: 3,535; javascript: 3,018; xml: 918; fortran: 664; cs: 573; ruby: 396
file content (88 lines) | stat: -rw-r--r-- 2,497 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// RUN: mlir-opt %s --convert-scf-to-cf --convert-cf-to-llvm --convert-vector-to-llvm \
// RUN:             --convert-func-to-llvm --convert-arith-to-llvm | \
// RUN:   mlir-cpu-runner -e entry -entry-point-result=void \
// RUN:                   --shared-libs=%mlir_c_runner_utils | \
// RUN:   FileCheck %s --match-full-lines

func.func @addi_i1(%v1 : i1, %v2 : i1) {
  vector.print str "@addi_i1\n"
  %res = arith.addi %v1, %v2 : i1
  vector.print %res : i1
  return
}

func.func @addi() {
  // ------------------------------------------------
  // Test i1
  // ------------------------------------------------

  // addi on i1
  // addi(0, 1) : i1 = 1 : i1; addi(0, -1) : i1 = 1
  %false = arith.constant 0 : i1
  %true = arith.constant 1 : i1

  // CHECK-LABEL: @addi_i1
  // CHECK-NEXT:  1
  func.call @addi_i1(%false, %true) : (i1, i1) -> ()

  // CHECK-LABEL: @addi_i1
  // CHECK-NEXT:  1
  %true_based_on_non_zero_val = arith.constant -1 : i1
  func.call @addi_i1(%false, %true_based_on_non_zero_val) : (i1, i1) -> ()

  // ------------------------------------------------
  // TODO: Test i8, i16 etc..
  // ------------------------------------------------

  return
}

func.func @addui_extended_i1(%v1 : i1, %v2 : i1) {
  vector.print str "@addui_extended_i1\n"
  %res, %overflow = arith.addui_extended %v1, %v2 : i1, i1
  vector.print %res : i1
  vector.print %overflow : i1
  return
}

func.func @addi_extended() {
  // ------------------------------------------------
  // Test i1
  // ------------------------------------------------

  // addui_extended on i1
  // addui_extended 1 1 : i1 = 0, 1
  %true = arith.constant 1 : i1
  %false = arith.constant 0 : i1
  
  // CHECK-LABEL: @addui_extended_i1
  // CHECK-NEXT:  0
  // CHECK-NEXT:  1
  func.call @addui_extended_i1(%true, %true) : (i1, i1) -> ()

  // CHECK-LABEL: @addui_extended_i1
  // CHECK-NEXT:  1
  // CHECK-NEXT:  0
  func.call @addui_extended_i1(%true, %false) : (i1, i1) -> ()

  // CHECK-LABEL: @addui_extended_i1
  // CHECK-NEXT:  1
  // CHECK-NEXT:  0
  func.call @addui_extended_i1(%false, %true) : (i1, i1) -> ()

  // CHECK-LABEL: @addui_extended_i1
  // CHECK-NEXT:  0
  // CHECK-NEXT:  0
  func.call @addui_extended_i1(%false, %false) : (i1, i1) -> ()

  // ------------------------------------------------
  // TODO: Test i8, i16 etc.. 
  // ------------------------------------------------
  return
}

func.func @entry() {
  func.call @addi() : () -> ()
  func.call @addi_extended() : () -> ()
  return
}