zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-29 14:58:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Metal validation (#432)

Browse Source 
* tests clear metal validation

* add cpp test with metal validation to circleci

* nit
This commit is contained in:
Awni Hannun
2024-01-11 11:57:24 -08:00
committed by GitHub
parent 975e265f74
commit c9934fe8a4
10 changed files with 142 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/allocator_tests.cpp
View File

@@ -38,6 +38,4 @@ TEST_CASE("test large allocations") {
auto buffer = allocator::malloc(size);
allocator::free(buffer);
}
// Shouldn't be able to allocate an exabyte anytime soon.
CHECK_THROWS_AS(allocator::malloc(1ull << 60), std::runtime_error);
}

2
tests/arg_reduce_tests.cpp
View File

@@ -1,7 +1,5 @@
// Copyright © 2023 Apple Inc.
#include <iostream>
#include "doctest/doctest.h"
#include "mlx/mlx.h"

33
tests/metal_tests.cpp
View File

@@ -438,3 +438,36 @@ TEST_CASE("test metal matmul") {
CHECK(array_equal(out, full({3, 3, 2, 2}, 2.0f), Device::cpu).item<bool>());
}
}
TEST_CASE("test metal validation") {
// Run this test with Metal validation enabled
// METAL_DEVICE_WRAPPER_TYPE=1 METAL_DEBUG_ERROR_MODE=0 ./tests/tests \
// -tc="test metal validation" \
auto x = array({});
eval(exp(x));
auto y = array({});
eval(add(x, y));
eval(sum(x));
x = array({1, 2, 3});
y = array(0);
eval(gather(x, y, 0, {0}));
eval(gather(x, y, 0, {2}));
eval(gather(x, y, 0, {0}));
eval(gather(x, y, 0, {2}));
eval(scatter(x, y, array({2}), 0));
x = arange(0, -3, 1);
eval(x);
array_equal(x, array({})).item<bool>();
x = array({1.0, 0.0});
eval(argmax(x));
eval(scatter_max(array(1), {}, array(2), std::vector<int>{}));
}