zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

add cuda

Browse Source
This commit is contained in:
Awni Hannun
2025-10-20 09:51:30 -07:00
parent ec4c85d786
commit 73613c4d15
6 changed files with 45 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ops_tests.cpp
View File

@@ -2,7 +2,6 @@
// Required for using M_PI_2 in MSVC.
#define _USE_MATH_DEFINES
#include <cmath>
#include <numeric>
@@ -4038,6 +4037,7 @@ TEST_CASE("test fp8 conversion") {
array in({-1.125, -1.0, 0.0, 1.0, 1.125, 4.5, 448.0}, t);
auto in_fp8 = to_fp8(in);
auto out = from_fp8(in_fp8, t);
CHECK(array_equal(out, in).item<bool>());
}
array in({-1.125, -1.0, 0.0, 1.0, 1.125, 4.5, 448.0});