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

catch stream errors earlier to avoid aborts (#1801)

Browse Source
This commit is contained in:
Awni Hannun
2025-01-27 14:05:43 -08:00
committed by GitHub
parent 28091aa1ff
commit 2235dee906
3 changed files with 29 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/random_tests.cpp
View File

@@ -421,6 +421,9 @@ TEST_CASE("test random normal") {
}
TEST_CASE("test random multivariate_normal") {
// Scope switch to the cpu for SVDs
StreamContext sc(Device::cpu);
{
auto mean = zeros({3});
auto cov = eye(3);