File: multiplication.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 (119 lines) | stat: -rw-r--r-- 3,593 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
// 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 @mulsi_extended_i1(%v1 : i1, %v2 : i1) {
  vector.print str "@mulsi_extended_i1\n"
  %low, %high = arith.mulsi_extended %v1, %v2 : i1
  vector.print %low : i1
  vector.print %high : i1
  return
}

func.func @mulsi_extended_i8(%v1 : i8, %v2 : i8) {
  vector.print str "@mulsi_extended_i8\n"
  %low, %high = arith.mulsi_extended %v1, %v2 : i8
  vector.print %low : i8
  vector.print %high : i8
  return
}

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

  // mulsi_extended on i1, tests for overflow bit
  // mulsi_extended 1, 1 : i1 = (1, 0)
  %true = arith.constant true
  %false = arith.constant false

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

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

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

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

  // ------------------------------------------------
  // Test i8
  // ------------------------------------------------
  // mulsi extended versions, with overflow
  %c_100_i8 = arith.constant -100 : i8

  // mulsi_extended -100, -100 : i8 = (16, 39)
  // CHECK-LABEL: @mulsi_extended_i8
  // CHECK-NEXT:  16
  // CHECK-NEXT:  39
  func.call @mulsi_extended_i8(%c_100_i8, %c_100_i8) : (i8, i8) -> ()

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

func.func @mului_extended_i8(%v1 : i8, %v2 : i8) {
  vector.print str "@mului_extended_i8\n"
  %low, %high = arith.mului_extended %v1, %v2 : i8
  vector.print %low : i8
  vector.print %high : i8
  return
}

func.func @mului_extended() {
  // ------------------------------------------------
  // Test i8
  // ------------------------------------------------
  %c_n100_i8 = arith.constant -100 : i8
  %c_156_i8 = arith.constant 156 : i8

  // mului_extended -100, -100 : i8 = (16, 95)
  // and on equivalent representations (e.g. 156 === -100 (mod 256))

  // CHECK-LABEL: @mului_extended_i8
  // CHECK-NEXT:  16
  // CHECK-NEXT:  95
  func.call @mului_extended_i8(%c_n100_i8, %c_n100_i8) : (i8, i8) -> ()

  // CHECK-LABEL: @mului_extended_i8
  // CHECK-NEXT:  16
  // CHECK-NEXT:  95
  func.call @mului_extended_i8(%c_n100_i8, %c_156_i8) : (i8, i8) -> ()

  // CHECK-LABEL: @mului_extended_i8
  // CHECK-NEXT:  16
  // CHECK-NEXT:  95
  func.call @mului_extended_i8(%c_156_i8, %c_n100_i8) : (i8, i8) -> ()

  // CHECK-LABEL: @mului_extended_i8
  // CHECK-NEXT:  16
  // CHECK-NEXT:  95
  func.call @mului_extended_i8(%c_156_i8, %c_156_i8) : (i8, i8) -> ()

  // ------------------------------------------------
  // TODO: Test i1, i16, i32 etc.. 
  // ------------------------------------------------
  return
}

func.func @entry() {
  func.call @mulsi_extended() : () -> ()
  func.call @mului_extended() : () -> ()
  return
}