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

safely divide for 0 size inputs (#388)

Browse Source
This commit is contained in:
Awni Hannun
2024-01-07 00:19:54 -08:00
committed by GitHub
parent b34bf5d52b
commit c6d2878c1a
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ops_tests.cpp
View File

@@ -2314,4 +2314,4 @@ TEST_CASE("tensordot") {
65370.},
{3, 6});
CHECK(array_equal(z, expected).item<bool>());
}
}