zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 16:48:10 +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

8
mlx/backend/cuda/jit_module.cpp
View File

@@ -67,9 +67,11 @@ const std::string& cccl_dir() {
return path.string();
}
// Finally check the environment variable.
path = std::getenv("MLX_CCCL_DIR");
if (!path.empty() && std::filesystem::exists(path)) {
return path.string();
if (const char* env = std::getenv("MLX_CCCL_DIR"); env) {
path = env;
if (!path.empty() && std::filesystem::exists(path)) {
return path.string();
}
}
return std::string();
}();

1
mlx/mlx.h
View File

@@ -4,6 +4,7 @@
#include "mlx/array.h"
#include "mlx/backend/cuda/cuda.h"
#include "mlx/backend/gpu/available.h"
#include "mlx/backend/metal/metal.h"
#include "mlx/compile.h"
#include "mlx/device.h"