Skip to content

Commit 3c8e7cc

Browse files
authored
Support complex numbers in BLAS level 2 and 3 rules (#822)
* Complex symv! * Add rule for hemv! * Extend trmv! rule to complex numbers * Tests for complex symv, hemv, trmv * Extend gemm! rule to complex numbers (6 tests failing, not my fault TM) * Complex symm! and hemm! * syrk and herk rules for complex numbers * Complex trmm! * Complex trsm! * Add the tests! * Make formatter happy * Fix gemm test * Split BLAS level 3 tests into a separate set * Fix level 3 test * Formatting * conj(x) is a no-op for real arrays, so simplify code * Reduce size of hemm matrices to make failures less likely * Hopefully fix unlucky rng for hemm and trsm * Give each test 'group' its own rng * format * Try another seed * Fix CI? * Fix CI? * Fix CI? * Bump version Signed-off-by: Bruno Ploumhans <[email protected]> --------- Signed-off-by: Bruno Ploumhans <[email protected]>
1 parent 2f4b010 commit 3c8e7cc

File tree

7 files changed

+655
-424
lines changed

7 files changed

+655
-424
lines changed

.github/workflows/CI.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ jobs:
2929
'basic',
3030
'rrules/avoiding_non_differentiable_code',
3131
'rrules/blas',
32+
'rrules/blas_level_3',
3233
'rrules/builtins',
3334
'rrules/dispatch_doctor',
3435
'rrules/fastmath',

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "Mooncake"
22
uuid = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
33
authors = ["Will Tebbutt, Hong Ge, and contributors"]
4-
version = "0.4.176"
4+
version = "0.4.177"
55

66
[deps]
77
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

0 commit comments

Comments
 (0)