shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

[mlir][complex] Correctness check for complex.conj

Browse Source 
Add correctness check for complex.conj operation

Reviewed By: pifon2a

Differential Revision: https://reviews.llvm.org/D127377
This commit is contained in:
lewuathe 2022-06-09 11:11:44 +02:00 committed by Alexander Belyaev
parent 447c411fef
commit fff27d181c
1 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
mlir/test/Integration/Dialect/Complex/CPU/correctness.mlir
View File

@ -43,6 +43,11 @@ func.func @rsqrt(%arg: complex<f32>) -> complex<f32> {
func.return %sqrt : complex<f32>
}
func.func @conj(%arg: complex<f32>) -> complex<f32> {
%conj = complex.conj %arg : complex<f32>
func.return %conj : complex<f32>
}
// %input contains pairs of lhs, rhs, i.e. [lhs_0, rhs_0, lhs_1, rhs_1,...]
func.func @test_binary(%input: tensor<?xcomplex<f32>>,
%func: (complex<f32>, complex<f32>) -> complex<f32>) {
@ -216,5 +221,36 @@ func.func @entry() {
call @test_unary(%rsqrt_test_cast, %rsqrt_func)
: (tensor<?xcomplex<f32>>, (complex<f32>) -> complex<f32>) -> ()
// complex.conj test
%conj_test = arith.constant dense<[
(-1.0, -1.0),
// CHECK: -1.0
// CHECK-NEXT: 1.0
(-1.0, 1.0),
// CHECK-NEXT: -1.0
// CHECK-NEXT: -1.0
(0.0, 0.0),
// CHECK-NEXT: 0
// CHECK-NEXT: 0
(0.0, 1.0),
// CHECK-NEXT: 0
// CHECK-NEXT: -1.0
(1.0, -1.0),
// CHECK-NEXT: 1.0
// CHECK-NEXT: -1.0
(1.0, 0.0),
// CHECK-NEXT: 1.0
// CHECK-NEXT: 0
(1.0, 1.0)
// CHECK-NEXT: 1.0
// CHECK-NEXT: -1.0
]> : tensor<7xcomplex<f32>>
%conj_test_cast = tensor.cast %conj_test
: tensor<7xcomplex<f32>> to tensor<?xcomplex<f32>>
%conj_func = func.constant @conj : (complex<f32>) -> complex<f32>
call @test_unary(%conj_test_cast, %conj_func)
: (tensor<?xcomplex<f32>>, (complex<f32>) -> complex<f32>) -> ()
func.return
}