This is an archive of the discontinued LLVM Phabricator instance.

Differential D127377

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

Authored by Lewuathe on Jun 8 2022, 10:19 PM.

Details

Reviewers
pifon2a
Commits
rGfff27d181c0c: [mlir][complex] Correctness check for complex.conj
Summary

Add correctness check for complex.conj operation

Diff Detail

Event Timeline

Lewuathe created this revision.Jun 8 2022, 10:19 PM
Herald added a project: Restricted Project. · View Herald TranscriptJun 8 2022, 10:19 PM
Herald added subscribers: bzcheeseman, sdasgup3, wenzhicui and 20 others. · View Herald Transcript
Lewuathe requested review of this revision.Jun 8 2022, 10:19 PM
Herald added a project: Restricted Project. · View Herald TranscriptJun 8 2022, 10:19 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
Harbormaster completed remote builds in B168744: Diff 435425.Jun 8 2022, 10:40 PM
pifon2a accepted this revision.Jun 9 2022, 2:11 AM
This revision is now accepted and ready to land.Jun 9 2022, 2:11 AM
Closed by commit rGfff27d181c0c: [mlir][complex] Correctness check for complex.conj (authored by Lewuathe, committed by pifon2a). · Explain WhyJun 9 2022, 2:12 AM
This revision was automatically updated to reflect the committed changes.
pifon2a added a commit: rGfff27d181c0c: [mlir][complex] Correctness check for complex.conj.

Revision Contents

PathSize
mlir/
test/
Integration/
Dialect/
Complex/
CPU/
36 lines

Diff 435457

mlir/test/Integration/Dialect/Complex/CPU/correctness.mlir

Show All 37 Linesfunc.func @tanh(%arg: complex<f32>) -> complex<f32> {
func.return %tanh : complex<f32> func.return %tanh : complex<f32>
} }
func.func @rsqrt(%arg: complex<f32>) -> complex<f32> { func.func @rsqrt(%arg: complex<f32>) -> complex<f32> {
%sqrt = complex.rsqrt %arg : complex<f32> %sqrt = complex.rsqrt %arg : complex<f32>
func.return %sqrt : 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,...] // %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.func @test_binary(%input: tensor<?xcomplex<f32>>,
%func: (complex<f32>, complex<f32>) -> complex<f32>) { %func: (complex<f32>, complex<f32>) -> complex<f32>) {
%c0 = arith.constant 0 : index %c0 = arith.constant 0 : index
%c1 = arith.constant 1 : index %c1 = arith.constant 1 : index
%c2 = arith.constant 2 : index %c2 = arith.constant 2 : index
%size = tensor.dim %input, %c0: tensor<?xcomplex<f32>> %size = tensor.dim %input, %c0: tensor<?xcomplex<f32>>
▲ Show 20 Lines • Show All 157 Lines • ▼ Show 20 Linesfunc.func @entry() {
]> : tensor<7xcomplex<f32>> ]> : tensor<7xcomplex<f32>>
%rsqrt_test_cast = tensor.cast %rsqrt_test %rsqrt_test_cast = tensor.cast %rsqrt_test
: tensor<7xcomplex<f32>> to tensor<?xcomplex<f32>> : tensor<7xcomplex<f32>> to tensor<?xcomplex<f32>>
%rsqrt_func = func.constant @rsqrt : (complex<f32>) -> complex<f32> %rsqrt_func = func.constant @rsqrt : (complex<f32>) -> complex<f32>
call @test_unary(%rsqrt_test_cast, %rsqrt_func) call @test_unary(%rsqrt_test_cast, %rsqrt_func)
: (tensor<?xcomplex<f32>>, (complex<f32>) -> complex<f32>) -> () : (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 func.return
} }