zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-20 09:33:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix complex power on Metal (#1460)

Browse Source
This commit is contained in:
Awni Hannun
2024-10-06 19:58:30 -07:00
committed by GitHub
parent e4534dac17
commit 95d04805b3
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/ops_tests.cpp
View File

@@ -2553,6 +2553,13 @@ TEST_CASE("test power") {
auto out = (power(array(a), array(b))).item<complex64_t>();
CHECK(abs(out.real() - expected.real()) < 1e-7);
CHECK(abs(out.imag() - expected.imag()) < 1e-7);
a = complex64_t{-1.2, 0.1};
b = complex64_t{2.2, 0.0};
expected = std::pow(a, b);
out = (power(array(a), array(b))).item<complex64_t>();
CHECK(abs(out.real() - expected.real()) < 1e-6);
CHECK(abs(out.imag() - expected.imag()) < 1e-6);
}
TEST_CASE("test where") {