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

Debug cuda conv (#2662)

Browse Source 
* use t4

* use t4
This commit is contained in:
Awni Hannun
2025-10-10 16:12:47 -07:00
committed by GitHub
parent 630350ad3e
commit 226a1d24e0
3 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/cuda/cudnn_utils.cpp
View File

@@ -210,6 +210,9 @@ std::optional<cudnn_frontend::ExecutionPlan> find_cudnn_plan_from_op_graph(
Dtype dtype,
cudnn_frontend::OperationGraph& op_graph) {
auto engine_configs = get_cudnn_engine_configs(backend_type, dtype, op_graph);
if (engine_configs.empty()) {
return std::nullopt;
}
return find_cudnn_plan_from_engine_configs(handle, engine_configs, op_graph);
}