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

Fix compilation

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-08-18 22:46:21 -07:00
parent d2ae81b413
commit 14efd9c35a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/cuda/custom_kernel.cpp
View File

@ -334,7 +334,7 @@ void CustomKernel::eval_gpu(
args.append_ndim(in.strides()); args.append_ndim(in.strides());
} }
if (shape_info.ndim) { if (shape_info.ndim) {
args.append(in.ndim()); args.append<int32_t>(in.ndim());
} }
} }
for (auto& out : outputs) { for (auto& out : outputs) {

2
mlx/backend/metal/custom_kernel.cpp
View File

@ -172,7 +172,7 @@ std::string write_template(
return template_def.str(); return template_def.str();
} }
MetalKernelFunction metal_kernel( CustomKernelFunction metal_kernel(
const std::string& name, const std::string& name,
const std::vector<std::string>& input_names, const std::vector<std::string>& input_names,
const std::vector<std::string>& output_names, const std::vector<std::string>& output_names,