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

Run CPP tests for CUDA build in CI (#2544)

Browse Source
This commit is contained in:
Cheng
2025-08-27 08:06:46 +09:00
committed by GitHub
parent 5458d43247
commit a9bac3d9e5
6 changed files with 29 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/scheduler_tests.cpp
View File

@@ -16,7 +16,7 @@ TEST_CASE("test stream management") {
CHECK_NE(s1, s2);
// Check that default streams have the correct devices
if (metal::is_available()) {
if (gpu::is_available()) {
auto s_gpu = default_stream(Device::gpu);
CHECK_EQ(s_gpu.device, Device::gpu);
} else {
@@ -28,7 +28,7 @@ TEST_CASE("test stream management") {
s_cpu = new_stream(Device::cpu);
CHECK_EQ(s_cpu.device, Device::cpu);
if (metal::is_available()) {
if (gpu::is_available()) {
auto s_gpu = new_stream(Device::gpu);
CHECK_EQ(s_gpu.device, Device::gpu);
} else {