zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix x86 tests (#1691)

Browse Source 
* fix x86 tests

* comment
This commit is contained in:
Awni Hannun
2024-12-11 07:47:18 -08:00
committed by GitHub
parent 4f9b60dd53
commit f3dfa36a3a
4 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/autograd_tests.cpp
View File

@@ -437,10 +437,10 @@ TEST_CASE("test op vjps") {
// Test erf
{
auto out = vjp([](array in) { return erf(in); }, array(inf), array(1.0f));
CHECK_EQ(out.second.item<float>(), 0.0f);
CHECK_EQ(out.second.item<float>(), doctest::Approx(0.0f));
out = vjp([](array in) { return erf(in); }, array(-inf), array(2.0f));
CHECK_EQ(out.second.item<float>(), 0.0f);
CHECK_EQ(out.second.item<float>(), doctest::Approx(0.0f));
out = vjp([](array in) { return erf(in); }, array(0.0f), array(1.0f));
CHECK_EQ(out.second.item<float>(), static_cast<float>(M_2_SQRTPI));