mlir/test/Dialect/Complex/ops.mlir

   1 // RUN: mlir-opt %s | mlir-opt | FileCheck %s
   2 // RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s
   3
   4
   5 // CHECK-LABEL: func @ops(
   6 // CHECK-SAME:            %[[F:.*]]: f32) {
   7 func.func @ops(%f: f32) {
   8   // CHECK: complex.constant [1.{{.*}}, -1.{{.*}}] : complex<f64>
   9   %cst_f64 = complex.constant [0.1, -1.0] : complex<f64>
  10
  11   // CHECK: complex.constant [1.{{.*}} : f32, -1.{{.*}} : f32] : complex<f32>
  12   %cst_f32 = complex.constant [0.1 : f32, -1.0 : f32] : complex<f32>
  13
  14   // CHECK: complex.constant [true, false] : complex<i1>
  15   %cst_i1 = complex.constant [1 : i1, 0 : i1] : complex<i1>
  16
  17   // CHECK: %[[C:.*]] = complex.create %[[F]], %[[F]] : complex<f32>
  18   %complex = complex.create %f, %f : complex<f32>
  19
  20   // CHECK: complex.re %[[C]] : complex<f32>
  21   %real = complex.re %complex : complex<f32>
  22
  23   // CHECK: complex.im %[[C]] : complex<f32>
  24   %imag = complex.im %complex : complex<f32>
  25
  26   // CHECK: complex.abs %[[C]] : complex<f32>
  27   %abs = complex.abs %complex : complex<f32>
  28
  29   // CHECK: complex.add %[[C]], %[[C]] : complex<f32>
  30   %sum = complex.add %complex, %complex : complex<f32>
  31
  32   // CHECK: complex.cos %[[C]] : complex<f32>
  33   %cos = complex.cos %complex : complex<f32>
  34
  35   // CHECK: complex.div %[[C]], %[[C]] : complex<f32>
  36   %div = complex.div %complex, %complex : complex<f32>
  37
  38   // CHECK: complex.eq %[[C]], %[[C]] : complex<f32>
  39   %eq = complex.eq %complex, %complex : complex<f32>
  40
  41   // CHECK: complex.exp %[[C]] : complex<f32>
  42   %exp = complex.exp %complex : complex<f32>
  43
  44   // CHECK: complex.expm1 %[[C]] : complex<f32>
  45   %expm1 = complex.expm1 %complex : complex<f32>
  46
  47   // CHECK: complex.log %[[C]] : complex<f32>
  48   %log = complex.log %complex : complex<f32>
  49
  50   // CHECK: complex.log1p %[[C]] : complex<f32>
  51   %log1p = complex.log1p %complex : complex<f32>
  52
  53   // CHECK: complex.mul %[[C]], %[[C]] : complex<f32>
  54   %prod = complex.mul %complex, %complex : complex<f32>
  55
  56   // CHECK: complex.neg %[[C]] : complex<f32>
  57   %neg = complex.neg %complex : complex<f32>
  58
  59   // CHECK: complex.neq %[[C]], %[[C]] : complex<f32>
  60   %neq = complex.neq %complex, %complex : complex<f32>
  61
  62   // CHECK: complex.sign %[[C]] : complex<f32>
  63   %sign = complex.sign %complex : complex<f32>
  64
  65   // CHECK: complex.sin %[[C]] : complex<f32>
  66   %sin = complex.sin %complex : complex<f32>
  67
  68   // CHECK: complex.sub %[[C]], %[[C]] : complex<f32>
  69   %diff = complex.sub %complex, %complex : complex<f32>
  70
  71   // CHECK: complex.tanh %[[C]] : complex<f32>
  72   %tanh = complex.tanh %complex : complex<f32>
  73
  74   // CHECK: complex.pow %[[C]], %[[C]] : complex<f32>
  75   %pow = complex.pow %complex, %complex : complex<f32>
  76
  77   // CHECK: complex.sqrt %[[C]] : complex<f32>
  78   %sqrt = complex.sqrt %complex : complex<f32>
  79
  80   // CHECK: complex.rsqrt %[[C]] : complex<f32>
  81   %rsqrt = complex.rsqrt %complex : complex<f32>
  82
  83   // CHECK: complex.atan2 %[[C]], %[[C]] : complex<f32>
  84   %atan2 = complex.atan2 %complex, %complex : complex<f32>
  85
  86   // CHECK: complex.tan %[[C]] : complex<f32>
  87   %tan = complex.tan %complex : complex<f32>
  88
  89   // CHECK: complex.bitcast %[[C]]
  90   %i64 = complex.bitcast %complex : complex<f32> to i64
  91
  92   return
  93 }