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

fix multi-output with compile

Browse Source
This commit is contained in:
Awni Hannun
2024-01-17 06:55:11 -08:00
parent ed4d867092
commit 390e50b163
5 changed files with 39 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/compile_tests.cpp
View File

@@ -1,5 +1,6 @@
// Copyright © 2023 Apple Inc.
#include <iostream>
#include "doctest/doctest.h"
#include "mlx/mlx.h"
@@ -99,9 +100,9 @@ TEST_CASE("test nested compile") {
TEST_CASE("test enable and disable compile") {
CHECK_THROWS(compile(nullptr));
disable_compiler();
disable_compile();
compile(nullptr);
enable_compiler();
enable_compile();
CHECK_THROWS(compile(nullptr));
}