1612993788
The op used to support only float element types. This was inconsistent with `ConstantOp::isBuildableWith`, which allows integer element types. The complex type allows any float/integer element type. Note: The other complex dialect ops do not support non-float element types yet. The main purpose of this change to fix `Tensor/canonicalize.mlir`, which is currently failing when verifying the IR after each pattern application (#74270). ``` within split at mlir/test/Dialect/Tensor/canonicalize.mlir:231 offset :8:15: error: 'complex.constant' op result #0 must be complex type with floating-point elements, but got 'complex<i32>' %complex1 = tensor.extract %c1[] : tensor<complex<i32>> ^ within split at mlir/test/Dialect/Tensor/canonicalize.mlir:231 offset :8:15: note: see current operation: %0 = "complex.constant"() <{value = [1 : i32, 2 : i32]}> : () -> complex<i32> "func.func"() <{function_type = () -> tensor<3xcomplex<i32>>, sym_name = "extract_from_elements_complex_i"}> ({ %0 = "complex.constant"() <{value = [1 : i32, 2 : i32]}> : () -> complex<i32> %1 = "arith.constant"() <{value = dense<(3,2)> : tensor<complex<i32>>}> : () -> tensor<complex<i32>> %2 = "arith.constant"() <{value = dense<(1,2)> : tensor<complex<i32>>}> : () -> tensor<complex<i32>> %3 = "tensor.extract"(%1) : (tensor<complex<i32>>) -> complex<i32> %4 = "tensor.from_elements"(%0, %3, %0) : (complex<i32>, complex<i32>, complex<i32>) -> tensor<3xcomplex<i32>> "func.return"(%4) : (tensor<3xcomplex<i32>>) -> () }) : () -> () ```
94 lines
3.0 KiB
MLIR
94 lines
3.0 KiB
MLIR
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
|
|
// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s
|
|
|
|
|
|
// CHECK-LABEL: func @ops(
|
|
// CHECK-SAME: %[[F:.*]]: f32) {
|
|
func.func @ops(%f: f32) {
|
|
// CHECK: complex.constant [1.{{.*}}, -1.{{.*}}] : complex<f64>
|
|
%cst_f64 = complex.constant [0.1, -1.0] : complex<f64>
|
|
|
|
// CHECK: complex.constant [1.{{.*}} : f32, -1.{{.*}} : f32] : complex<f32>
|
|
%cst_f32 = complex.constant [0.1 : f32, -1.0 : f32] : complex<f32>
|
|
|
|
// CHECK: complex.constant [true, false] : complex<i1>
|
|
%cst_i1 = complex.constant [1 : i1, 0 : i1] : complex<i1>
|
|
|
|
// CHECK: %[[C:.*]] = complex.create %[[F]], %[[F]] : complex<f32>
|
|
%complex = complex.create %f, %f : complex<f32>
|
|
|
|
// CHECK: complex.re %[[C]] : complex<f32>
|
|
%real = complex.re %complex : complex<f32>
|
|
|
|
// CHECK: complex.im %[[C]] : complex<f32>
|
|
%imag = complex.im %complex : complex<f32>
|
|
|
|
// CHECK: complex.abs %[[C]] : complex<f32>
|
|
%abs = complex.abs %complex : complex<f32>
|
|
|
|
// CHECK: complex.add %[[C]], %[[C]] : complex<f32>
|
|
%sum = complex.add %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.cos %[[C]] : complex<f32>
|
|
%cos = complex.cos %complex : complex<f32>
|
|
|
|
// CHECK: complex.div %[[C]], %[[C]] : complex<f32>
|
|
%div = complex.div %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.eq %[[C]], %[[C]] : complex<f32>
|
|
%eq = complex.eq %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.exp %[[C]] : complex<f32>
|
|
%exp = complex.exp %complex : complex<f32>
|
|
|
|
// CHECK: complex.expm1 %[[C]] : complex<f32>
|
|
%expm1 = complex.expm1 %complex : complex<f32>
|
|
|
|
// CHECK: complex.log %[[C]] : complex<f32>
|
|
%log = complex.log %complex : complex<f32>
|
|
|
|
// CHECK: complex.log1p %[[C]] : complex<f32>
|
|
%log1p = complex.log1p %complex : complex<f32>
|
|
|
|
// CHECK: complex.mul %[[C]], %[[C]] : complex<f32>
|
|
%prod = complex.mul %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.neg %[[C]] : complex<f32>
|
|
%neg = complex.neg %complex : complex<f32>
|
|
|
|
// CHECK: complex.neq %[[C]], %[[C]] : complex<f32>
|
|
%neq = complex.neq %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.sign %[[C]] : complex<f32>
|
|
%sign = complex.sign %complex : complex<f32>
|
|
|
|
// CHECK: complex.sin %[[C]] : complex<f32>
|
|
%sin = complex.sin %complex : complex<f32>
|
|
|
|
// CHECK: complex.sub %[[C]], %[[C]] : complex<f32>
|
|
%diff = complex.sub %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.tanh %[[C]] : complex<f32>
|
|
%tanh = complex.tanh %complex : complex<f32>
|
|
|
|
// CHECK: complex.pow %[[C]], %[[C]] : complex<f32>
|
|
%pow = complex.pow %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.sqrt %[[C]] : complex<f32>
|
|
%sqrt = complex.sqrt %complex : complex<f32>
|
|
|
|
// CHECK: complex.rsqrt %[[C]] : complex<f32>
|
|
%rsqrt = complex.rsqrt %complex : complex<f32>
|
|
|
|
// CHECK: complex.atan2 %[[C]], %[[C]] : complex<f32>
|
|
%atan2 = complex.atan2 %complex, %complex : complex<f32>
|
|
|
|
// CHECK: complex.tan %[[C]] : complex<f32>
|
|
%tan = complex.tan %complex : complex<f32>
|
|
|
|
// CHECK: complex.bitcast %[[C]]
|
|
%i64 = complex.bitcast %complex : complex<f32> to i64
|
|
|
|
return
|
|
}
|